Python Package

pynvml

2022-12-07

pynvml 是一个用于获取 NVIDIA GPU 信息的 [[Python]] 库。

安装

Terminal window
1
pip install pynvml

使用

1
import pynvml
2


3
pynvml.nvmlInit()     # 初始化
4


5
pynvml.nvmlSystemGetDriverVersion() # 获取驱动版本信息
6
pynvml.nvmlDeviceGetCount()         # 获取显卡数量
7


8
pynvml.nvmlShutdown() # 退出
# Python
# Package
# NVIDIA