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()
handle = len(pynvml.nvmldevicegethandlebyindex(0))#
這裡的0是gpu id
meminfo =pynvml.nvmldevicegetmemoryinfo(handle)
print(meminfo.total) #
第二塊顯示卡總的視訊記憶體大小
print(meminfo.used)#
這裡是位元組bytes，所以要想得到以兆m為單位就需要除以1024**2
print(meminfo.free) #
第二塊顯示卡剩餘視訊記憶體大小
print(pynvml.nvmldevicegetcount())#
顯示有幾塊gpu

posted @

2019-01-19 17:49

小呆丶 閱讀(

...)

編輯收藏

python檢視顯示卡gpu資訊

需要使用pynvml庫 官網 現階段pip安裝的命令為 sudo pip install nvidia ml pyimport pynvml pynvml.nvmlinit 這裡的1是gpu id handle pynvml.nvmldevicegethandlebyindex 1 meminfo ...

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 或者是把上述路徑新增到環...