pynvml 查看GPU已使用的显存

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/82982693

Python查看GPU已使用的显存

版权声明：随意转载，不用告诉我，但链接到就行。 https://blog.csdn.net/u011094454/article/details/80774495

pip安装pynvml

pip install nvidia-ml-py3

•  

如果是Python2，则pip nvidia-ml-py2

之后可以在Python3 下使用了

import pynvml
pynvml.nvmlInit()
# 这里的0是GPU id
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.used)

GPU其他信息，可以查看API文档