pynvml

1
import pynvml
2

3
pynvml.nvmlInit()     # 初始化
4

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

8
pynvml.nvmlShutdown() # 退出