Python3 查看GPU信息

# -*- coding: utf-8 -*-
"""
Spyder Editor

This is a temporary script file.
"""
#pip install nvidia-ml-py3

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