diff --git a/lib/get_specs.py b/lib/get_specs.py index 176b67d..ec3ea87 100644 --- a/lib/get_specs.py +++ b/lib/get_specs.py @@ -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