pynvml 是一个用于获取 NVIDIA GPU 信息的 [[Python]] 库。
安装
pip install pynvml使用
import pynvml
pynvml.nvmlInit() # 初始化
pynvml.nvmlSystemGetDriverVersion() # 获取驱动版本信息pynvml.nvmlDeviceGetCount() # 获取显卡数量
pynvml.nvmlShutdown() # 退出pynvml 是一个用于获取 NVIDIA GPU 信息的 [[Python]] 库。
pip install pynvmlimport pynvml
pynvml.nvmlInit() # 初始化
pynvml.nvmlSystemGetDriverVersion() # 获取驱动版本信息pynvml.nvmlDeviceGetCount() # 获取显卡数量
pynvml.nvmlShutdown() # 退出