Debug CMP170
This commit is contained in:
parent
35104eff6a
commit
6170c98f7b
|
@ -290,7 +290,7 @@ def get_gpu_info():
|
|||
"mem_free": parts[11],
|
||||
"mem_used": parts[12]
|
||||
}
|
||||
print(xl_gpu_info)
|
||||
|
||||
try:
|
||||
pci_query = parts[2][parts[2].find(':')+1:]
|
||||
for index, valid_pci_dev in enumerate(valid_pci_dev_list):
|
||||
|
@ -308,10 +308,13 @@ def get_gpu_info():
|
|||
if bool(re.match(r'^[0-9]+$', parts[0])):
|
||||
|
||||
gpu_name = parts[1]
|
||||
print("p1" + gpu_name)
|
||||
if gpu_name == "NVIDIA Graphics Device" and parts[5] in GPU_ID_TO_NAME:
|
||||
gpu_name = GPU_ID_TO_NAME[parts[5]]
|
||||
print("p1" + gpu_name)
|
||||
|
||||
gpu_str = f"{len(lines)-1}x {gpu_name.strip()}"
|
||||
print(gpu_str)
|
||||
gpu_mem = round(int(filter_non_numeric(parts[4]).strip())/1024, 2)
|
||||
except Exception as e:
|
||||
nvml_err=True
|
||||
|
|
Loading…
Reference in New Issue