From f68e17634066fe51833c73a25fce4fe885a0afc5 Mon Sep 17 00:00:00 2001 From: empresa Date: Thu, 16 Oct 2025 01:57:30 +0700 Subject: [PATCH] dev --- lib/nvml.py | 4 ++++ 1 file changed, 4 insertions(+) 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)