python环境获取Nvidia显卡型号和驱动版本

GPUtil通过解析nvidia-smi获取显卡名称和驱动版本信息等信息。

测试环境

• python 3.6
• windows 10 (阅读源码，Linux环境应该也可以使用，未测试)

安装GPUtil

pip install gputil

使用

获取显卡型号，驱动版本。

In [1]: import GPUtil

In [2]: GPUtil.getFirstAvailable()
Out[2]: [0]

In [3]: GPUtil.getGPUs()
Out[3]: []

In [4]: z = _[0]

In [5]: z.name
Out[5]: 'GeForce RTX 2080'

In [6]: z.driver
Out[6]: '441.66'