pynvml模块使用方法

  • 2022-12-07 17:20:27
  • 开发
  • 998
  • shevechco

最近有需要使用nvml调用底层显卡信息的需求,所以查了下python有相关的模块pynvml可以使用,下面就直接看看一些基本使用的方法

首先我们安装这个模块

01.
pip install nvidia-ml-py3

然后使用时我们需要初始化该方法

01.
import pynvml
02.
pynvml.nvmlInit()

使用外以后也需要进行关闭,类似于析构方法

01.
pynvml.nvmlShutdown()

然后下面是收集的一些常用方法模块

01.
#!/usr/bin/python3
02.
#pip install nvidia-ml-py3
03.
import pynvml
04.
05.
pynvml.nvmlInit() # 初始化
06.
07.
#获取GPU卡驱动
08.
print(pynvml.nvmlSystemGetDriverVersion())
09.
10.
#获取显卡个数
11.
print(pynvml.nvmlDeviceGetCount())
12.
13.
#获取显卡句柄,gpu_id为0,1,2
14.
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
15.
16.
#获取显卡名字型号
17.
print(pynvml.nvmlDeviceGetName(handle))
18.
19.
#获取显卡内存信息
20.
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
21.
print(meminfo)
22.
#c_nvmlMemory_t(total: 12582912000 B, free: 12582912000 B, used: 0 B)
23.
24.
#获取显卡温度、风扇、电源
25.
print("核心温度: {} C".format(pynvml.nvmlDeviceGetTemperature(handle,0)))
26.
print("风扇转速: {}".format(pynvml.nvmlDeviceGetFanSpeed(handle)))
27.
print("电源状态: {}".format(pynvml.nvmlDeviceGetPowerState(handle)))
28.
29.
#获取GPU UUID
30.
print(pynvml.nvmlDeviceGetUUID(handle))
31.
32.
# 最后要关闭管理工具
33.
pynvml.nvmlShutdown()

打印如下

image.png

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:http://www.sulao.cn/post/900

相关推荐