-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgpu_parse.py
27 lines (24 loc) · 913 Bytes
/
gpu_parse.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from amdsmi import *
try:
amdsmi_init()
num_of_GPUs = len(amdsmi_get_processor_handles())
print("Number of GPUs ", num_of_GPUs)
if num_of_GPUs > 0:
handles = amdsmi_get_processor_handles()
print("----------------------------")
for i, handle in enumerate(handles):
print(f"GPU number {i} details: ")
type_of_GPU = amdsmi_get_processor_type(handle)
print("amdsmi_get_processor_type ",type_of_GPU)
device_id = amdsmi_get_gpu_device_uuid(handle)
print("amdsmi_get_gpu_device_uuid ",device_id)
driver_info = amdsmi_get_gpu_driver_info(handle)
print("amdsmi_get_gpu_driver_info ", driver_info)
print("----------------------------")
except AmdSmiException as e:
print(e)
finally:
try:
amdsmi_shut_down()
except AmdSmiException as e:
print(e)