主要内容转载自：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 状态查看的全部内容，更多相关深度学习中内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复