pynvml模块使用方法
- 2022-12-07 17:20:27
- 开发
- 998
- shevechco
最近有需要使用nvml调用底层显卡信息的需求,所以查了下python有相关的模块pynvml可以使用,下面就直接看看一些基本使用的方法
首先我们安装这个模块
01.pip install nvidia-ml-py3
然后使用时我们需要初始化该方法
01.import pynvml02.pynvml.nvmlInit()
使用外以后也需要进行关闭,类似于析构方法
01.pynvml.nvmlShutdown()
然后下面是收集的一些常用方法模块
01.#!/usr/bin/python302.#pip install nvidia-ml-py303.import pynvml04. 05.pynvml.nvmlInit() # 初始化06. 07.#获取GPU卡驱动08.print(pynvml.nvmlSystemGetDriverVersion())09. 10.#获取显卡个数11.print(pynvml.nvmlDeviceGetCount())12. 13.#获取显卡句柄,gpu_id为0,1,214.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 UUID30.print(pynvml.nvmlDeviceGetUUID(handle))31. 32.# 最后要关闭管理工具33.pynvml.nvmlShutdown()
打印如下
内容版权声明:除非注明,否则皆为本站原创文章。
转载注明出处:http://www.sulao.cn/post/900