diff --git a/lib/nvml.py b/lib/nvml.py index bcb1d31..1eca64d 100644 --- a/lib/nvml.py +++ b/lib/nvml.py @@ -100,9 +100,13 @@ def init(gpu_specs_file=None, allow_hive_binaries=True): gpu_uuid = pynvml.nvmlDeviceGetUUID(gpu_handle) gpu_name = pynvml.nvmlDeviceGetName(gpu_handle) + print(gpu_name) gpu_device_id = hex(pynvml.nvmlDeviceGetPciInfo(gpu_handle).pciDeviceId).upper() + print(gpu_device_id) + print(constants.GPU_ID_TO_NAME) if gpu_name == "NVIDIA Graphics Device" and gpu_device_id in constants.GPU_ID_TO_NAME: gpu_name = constants.GPU_ID_TO_NAME[gpu_device_id] + print(gpu_name) gpu_name_list.append(gpu_name)