【碎片知识】通过python3查看GPU信息

1.需要下载一个安装库

pip install nvidia-ml-py3  
pip install nvidia-ml-py 

pip install nvidia-ml-py3  #这是python3的，好多人没写这里，【坑死我了】！！
pip install nvidia-ml-py   #这是python2的

 

import pynvml
pynvml.nvmlInit()
# 这里的0是GPU id 【这里又是一个大坑】
handle = pynvml.nvmlDeviceGetHandleByIndex(0)  # 我的GPU从0开始的，很多都是从1开始的
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total) #第二块显卡总的显存大小
print(meminfo.used)#这里是字节bytes，所以要想得到以兆M为单位就需要除以1024**2
print(meminfo.free) #第二块显卡剩余显存大小

>>>:

4294967296
567746560
3727220736

 

以上的内容，【完全没有用】，我是想实时查看gpu的信息，目前只能通过外部的linux指令查看了，记得在Linux系统

↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓↓

最常用的参数是 -n， 后面指定是每多少秒来执行一次命令。

每隔10s显示一次显存情况

watch -n 10 nvidia-smi