python获取计算机上英伟达独立显卡信息

ric-hman

已于 2023-11-20 10:20:58 修改

阅读量507

点赞数

文章标签： python

于 2023-11-17 16:56:05 首次发布

本文链接：https://blog.csdn.net/qq_39142273/article/details/134466183

版权

本文介绍了如何通过Python的pynvml库来获取计算机上的英伟达独立显卡信息。内容包括安装pynvml包、使用nvml绑定和nvidia_smi API，以及注意NVML权限问题，特别是在没有管理员权限时可能遇到的GPU性能计数器查询限制。同时，提供了查阅权限状态和解决权限错误的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装pynvml包

# 错误指令
（×） pip install nvidia-ml-py
（×）pip install nvidia-ml-py3
# 通过上面两个指令安装包后运行下面的程序会报错；通过下面指令安装后可以运行
pip install pynvml

你可以使用底层的nvml绑定:

from pynvml import *
nvmlInit() #初始化
print("Driver Version:", nvmlSystemGetDriverVersion()) #驱动版本
deviceCount = nvmlDeviceGetCount() #显卡数量
for i in range(deviceCount):
    handle = nvmlDeviceGetHandleByIndex(i)
    print("Device", i, ":", nvmlDeviceGetName(handle))
nvmlShutdown()

或者更高级的nvidia_smi API:

from pynvml.smi import nvidia_smi
nvsmi = nvidia_smi.getInstance()
nvsmi.DeviceQuery('memory.free, memory.total')


from pynvml.smi import nvidia_smi
nvsmi = nvidia_smi.getInstance()
print(nvsmi.DeviceQuery('--help-query-gpu'), end='\n')

Functions

Pytho

最低0.47元/天解锁文章