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]: [<GPUtil.GPUtil.GPU at 0x20b21936c18>]
In [4]: z = _[0]
In [5]: z.name
Out[5]: 'GeForce RTX 2080'
In [6]: z.driver
Out[6]: '441.66'