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