admin管理员组

文章数量:1130349

1.查看CUDA版本以及GPU信息

nvidia-smi

2.使用pynvml查看GPU使用情况

首先安装nvidia-ml-py包

pip install nvidia-ml-py

代码:

import pynvml
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 指定显卡号
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total/1024**2) #总的显存大小(float)
print(meminfo.used/1024**2)  #已用显存大小(float)
print(meminfo.free/1024**2)  #剩余显存大小(float)
print(pynvml.nvmlDeviceGetCount())#显示有几块GPU

3.使用GPUti实时查看GPU状况

pip install gputil

代码

import GPUtil
GPUtil.showUtilization()

ort GPUtil
GPUtil.showUtilization()


![image-20220409175419006](https://img-blog.csdnimg/img_convert/6e19aa4be89558001c67f6a742b6c28c.png)

1.查看CUDA版本以及GPU信息

nvidia-smi

2.使用pynvml查看GPU使用情况

首先安装nvidia-ml-py包

pip install nvidia-ml-py

代码:

import pynvml
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 指定显卡号
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total/1024**2) #总的显存大小(float)
print(meminfo.used/1024**2)  #已用显存大小(float)
print(meminfo.free/1024**2)  #剩余显存大小(float)
print(pynvml.nvmlDeviceGetCount())#显示有几块GPU

3.使用GPUti实时查看GPU状况

pip install gputil

代码

import GPUtil
GPUtil.showUtilization()

ort GPUtil
GPUtil.showUtilization()


![image-20220409175419006](https://img-blog.csdnimg/img_convert/6e19aa4be89558001c67f6a742b6c28c.png)

本文标签: 情况WindowsGPU