python檢視顯示卡gpu資訊

需要使用pynvml庫

官網：

現階段pip安裝的命令為：

sudo pip install nvidia-ml-py

import pynvml

pynvml.nvmlinit()
# 這裡的1是gpu id
handle = pynvml.nvmldevicegethandlebyindex(1)
meminfo = pynvml.nvmldevicegetmemoryinfo(handle)
print(meminfo.total) #第二塊顯示卡總的視訊記憶體大小
print(meminfo.used)#這裡是位元組bytes，所以要想得到以兆m為單位就需要除以1024**2
print(meminfo.free) #第二塊顯示卡剩餘視訊記憶體大小

經過計算，總的現存大小11721506816 / 1024 /1024 = 11178m， 已使用5333057536 / 1024 /1024 =5086m，然後在ubuntu終端中輸入nvidia-smi檢視顯示卡資訊：

>>> print "driver version:", nvmlsystemgetdriverversion()#顯示卡驅動版本

driver version: 304.00

>>> devicecount = nvmldevicegetcount()#幾塊顯示卡

>>> for i in range(devicecount):

... handle = nvmldevicegethandlebyindex(i)

... print "device", i, ":", nvmldevicegetname(handle) #具體是什麼顯示卡

...device 0 : tesla c2070

>>> nvmlshutdown()

python 檢視顯示卡gpu資訊

需要安裝pynvml庫。pip安裝的命令為 pip install nvidia ml py 如下 usr bin env python coding utf 8 time 2019 1 19 17 40 author xiaodai import pynvml pynvml.nvmlinit ha...

python 檢視顯示卡gpu資訊

需要安裝pynvml庫。pip安裝的命令為 pip install nvidia ml py3 如下 usr bin env python coding utf 8 time 2019 1 19 17 40 author xiaodai import pynvml pynvml.nvmlinit h...

Linux Windows檢視GPU資訊

nvidia smiwindows系統下 需要進入到 c program files nvidia corporation nvsmi 此路徑下，再輸入上述命令，即 cd c program files nvidia corporation nvsmi nvidia smi 或者是把上述路徑新增到環...