Python：使用pynvml获取和监控 NVIDIA GPU 设备的信息

彭世瑜

已于 2023-11-18 11:46:22 修改

阅读量1k

点赞数

分类专栏： Python 文章标签： python 开发语言

于 2023-11-18 11:40:08 首次发布

本文为博主原创文章，欢迎转载，请注明出处

本文链接：https://blog.csdn.net/mouday/article/details/134476278

版权

Python 专栏收录该内容

614 篇文章

订阅专栏

pynvml是一个Python包，用于与NVIDIA的NVML库交互，提供获取和监控GPU设备信息的功能，包括状态、显存、温度和功耗等。文章介绍了安装步骤和基本用法，如初始化、获取GPU数量和信息，以及资源清理。

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

pynvml 是一个 Python 包，用于与 NVIDIA 的管理库 NVML（NVIDIA Management Library）进行交互，以获取和监控 NVIDIA GPU 设备的信息。它提供了一组函数和类，可以查询 GPU 设备的状态、显存使用情况、温度、功耗等信息。

文档

https://developer.nvidia.com/
https://pypi.org/project/pynvml/

安装

需要安装 NVIDIA 驱动程序和 CUDA 工具包

pip install pynvml

示例

import pynvml

# 初始化 pynvml
pynvml.nvmlInit()

# 获取 GPU 数量
print(pynvml.nvmlDeviceGetCount())

# 获取GPU信息
for i in range(deviceCount):
   handle = pynvml.nvmlDeviceGetHandleByIndex(i)
   info = pynvml.nvmlDeviceGetMemoryInfo(handle)
   gpu_name = pynvml.nvmlDeviceGetName(handle)

# 清理和释放资源
pynvml.nvmlShutdown()