clore
/
hosting
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

dev

This commit is contained in:
empresa 2025-10-16 01:57:30 +07:00
parent bae3d395f9
commit f68e176340
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/nvml.py
View File

@ -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)