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()

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()

版权声明:本文标题:Windows下GPU使用情况查看 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://it.en369.cn/jiaocheng/1754431119a2685494.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。


发表评论