【监控】python查看显卡资源使用情况

使用第三方库

需要使用pynvml库

官网:https://pythonhosted.org/nvidia-ml-py/

下载文件地址:https://pypi.org/project/nvidia-ml-py/#history

现阶段pip安装的命令为:

pip install nvidia-ml-py

python查看代码如下

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

问题

在python2下面可以正常运行,但是python3下面由于python自身原因导致会print报错,修改文件:python3.6/site-packages/pynvml.py如下:
【监控】python查看显卡资源使用情况_第1张图片
即可正常使用;

参考:
https://blog.csdn.net/u013066730/article/details/82756379

你可能感兴趣的:(GPU,python,python,gpu,linux)