windows和linux下查看或获取GPU的相关信息(附代码)

windows下查GPU相关信息，在命令行输入 nvidia-smi,即可得到，如下图：

 用如下python程序可以再linux和windows下输出显卡相关信息，例如显存占用等等，如下所示：

# pip install nvidia-ml-py or python3 -m pip install nvidia-ml-py
# home info: https://pypi.org/project/nvidia-ml-py/

import pynvml
pynvml.nvmlInit()
print("gpus total: ",pynvml.nvmlDeviceGetCount())                       #显示有几块GPU
print(f"driver Version: {pynvml.nvmlSystemGetDriverVersion()}")         #版本信息

deviceHandle = pynvml.nvmlDeviceGetHandleByIndex(0)                     # 指定显卡号
gpuMemoryInfo = pynvml.nvmlDeviceGetMemoryInfo(deviceHandle)
print("gpu memory total:",gpuMemoryInfo.total/1024**2, "MB")            #总的显存大小
print("gpu busy memory total:",gpuMemoryInfo.used/1024**2, "MB")        #已用显存大小
print("gpu free memory total:",gpuMemoryInfo.free/1024**2, "MB")        #剩余显存大小
pynvml.nvmlShutdown()
#linux下还可用 gpustat --w 获取相关gpu信息
#pip install gpustat
#watch --color -n1 gpustat -cpu
#可在linux bash中执行命令: lspci | grep -i vga 查看基本显卡信息
#或者在linux中用 sudo pip3 install gputil 也可以
#下面两句命令可以让你升级pip,以便支持相关的模块安装:
#pip install --upgrade setuptools
#python -m pip install --upgrade pip
print("running end")