深度学习中 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。