python中查看显卡状态

pip安装pynvml模块

import pynvml
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 0表示第一块显卡
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total/1024**2) #第二块显卡总的显存大小
print(meminfo.used/1024**2)  #这里是字节bytes，所以要想得到以兆M为单位就需要除以1024**2
print(meminfo.free/1024**2)  #第二块显卡剩余显存大小