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