diff --git a/lib/nvml.py b/lib/nvml.py index b44b792..1606767 100644 --- a/lib/nvml.py +++ b/lib/nvml.py @@ -104,8 +104,8 @@ def init(gpu_specs_file=None, allow_hive_binaries=True): print(pynvml.nvmlDeviceGetName(gpu_handle)) print("device_id") pci_info = pynvml.nvmlDeviceGetPciInfo(gpu_handle) - print(pci_info.pciDeviceId) - + print(hex(pci_info.pciDeviceId)) + gpu_name_list.append(pynvml.nvmlDeviceGetName(gpu_handle)) if not f"{i}-{gpu_uuid}" in parsed_specs_keys: