欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

【显卡】python获取GPU(显卡)资源问题

程序员文章站 2022-03-11 14:51:25
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,修改如下:
【显卡】python获取GPU(显卡)资源问题
修改完就可以支持python3。

本文地址:https://blog.csdn.net/m0_37661841/article/details/109256680