python環境獲取Nvidia顯示卡型號和驅動版本
阿新 • • 發佈：2021-01-19
技術標籤：Python
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'