python-----查看顯卡gpu信息

需要安裝pynvml庫。

下載地址為：https://pypi.org/project/nvidia-ml-py/#history

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