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()
发布者：admin，转转请注明出处：http://www.yc00.com/web/1754431119a5159745.html
评论列表（0条）