安装pynvml包
# 错误指令
(×) pip install nvidia-ml-py
(×)pip install nvidia-ml-py3
# 通过上面两个指令安装包后运行下面的程序会报错;通过下面指令安装后可以运行
pip install pynvml
你可以使用底层的nvml绑定:
from pynvml import *
nvmlInit() #初始化
print("Driver Version:", nvmlSystemGetDriverVersion()) #驱动版本
deviceCount = nvmlDeviceGetCount() #显卡数量
for i in range(deviceCount):
handle = nvmlDeviceGetHandleByIndex(i)
print("Device", i, ":", nvmlDeviceGetName(handle))
nvmlShutdown()
或者更高级的nvidia_smi API:
from pynvml.smi import nvidia_smi
nvsmi = nvidia_smi.getInstance()
nvsmi.DeviceQuery('memory.free, memory.total')
from pynvml.smi import nvidia_smi
nvsmi = nvidia_smi.getInstance()
print(nvsmi.DeviceQuery('--help-query-gpu'), end='\n')
Functions
Pytho