【显卡】python获取GPU(显卡)资源问题
程序员文章站
2022-06-22 09:34:34
nvidia-ml-py安装版本:375.53.1pip install nvidia-ml-py使用import pynvml# 初始化模块pynvml.nvmlInit()# 获取0号位GPUgpu = punvml.nvmlDeviceGeHandleByIndex(0)gpu_info = pynvml.nvmlDeviceGetMemoryInfo(gpu)获取的gpu_info有三个字段信息,获取方式如下gpu_info.total # 总量gpu_info.used...
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。
本文地址:https://blog.csdn.net/m0_37661841/article/details/109256680