【显卡】python查看GPU(显卡)资源

nvidia-ml-py安装

版本：375.53.1
pip install nvidia-ml-py

使用

import pynvml
# 初始化模块
pynvml.nvmlInit()
# 获取0号位GPU
gpu = punvml.nvmlDeviceGeHandleByIndex(0)
gpu_info = pynvml.nvmlDeviceGetMemoryInfo(gpu)

获取的gpu_info有三个字段信息，获取方式如下

gpu_info.total # 总量
gpu_info.used # 使用量
gpu_info.free # 空余量

问题

SyntaxError: Missing parentheses in call to 'print'. Did you mean print(c_count.value)?

报错原因是这个版本支持python2，有一个地方print方法还是python2不支持python3，修改如下：

修改完就可以支持python3。