【监控】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