深度学习中 GPU 状态查看

110 阅读 0 评论 73 点赞

我是靠谱客的博主虚幻茉莉，最近开发中收集的这篇文章主要介绍深度学习中 GPU 状态查看，觉得挺不错的，现在分享给大家，希望可以做个参考。

主要内容转载自：https://blog.csdn.net/lien0906/article/details/78863118

一、显存查看

方法 1：nvidia-smi

Nvidia 显卡命令行管理套件，显示效果如下

方法 2：gpustat

使用 pip install gpustat 即可安装，基于 nvidia-smi，可以提供更美观简洁的展示，结合 watch 命令，可以动态实时监控 GPU 的使用情况。

watch --color -n1 gpustat -cpu

二、无进程显存占用

有时用上面的命令会发现显存被占用但是没有进程

用 fuser -v /dev/nvidia* 可以显示如下 top 不会显示的进程

只要把对应 GPU 中的进程 kill 掉就可以清空对应 GPU 的显存

三、jupyter notebook 和 PyTorch

在 jupyter notebook 中通过 python notebook 使用 PyTorch 时，占用的显存不会因为关闭这个 notebook 而清除。

所以如果希望清空 notebook 占用的显存，需要 clear 或者 shutdown 这个 notebook。

以上就是虚幻茉莉为你收集整理的深度学习中 GPU 状态查看的全部内容，希望文章能够帮你解决深度学习中 GPU 状态查看所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。