python获取gpu信息

安装包nvidia-ml-py

pip install nvidia-ml-py==11.47.66

一定要带上版本,不然会下载版本为375.53.1,支持的版本为python2,需要去修改其中的print代码.

获取信息

import pynvml

pynvml.nvmlInit()  # 必须先初始化
#1.获取显卡驱动版本
Driver_Version = pynvml.nvmlSystemGetDriverVersion()
#2.获取gpu数量
deviceCount =pynvml.nvmlDeviceGetCount()
#3.获取第i个显卡的具体性信息:型号
handle = pynvml.nvmlDeviceGetHandleByIndex(i)
#4.获取第i个显卡的显存大小,已使用,剩余
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total) #第二块显卡总的显存大小
print(meminfo.used)#这里是字节bytes，所以要想得到以兆M为单位就需要除以1024**2
print(meminfo.free) #第二块显卡剩余显存大小

#结束
pynvml.nvmlShutdown()