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(
)handle = 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

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 py 如下 usr bin env python coding utf 8 time 2019 1 19 17 40 author xiaodai import pynvml pynvml.nvmlinit ha...

Linux Windows檢視GPU資訊

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