实时查看GPU显存占用、功耗、进程状态

简介

当运行深度学习命令时，我们总想查看GPU占用了多少显存，以便及时调整模型的参数。我之前总是’nvidia-smi’，然后一直↑+enter，一直盯着屏幕看显存占用情况。

后来感觉太麻烦了，还要一直手动点击，真是费劲。现在

我推荐使用以下两种方法查看显存占用情况。

方法 1：使用 nvitop

我推荐使用这个命令，感觉可视化做的挺好的，不仅能看GPU显存，还能查看内存使用。

nvitop 是类似 top 命令的工具，可以实时监测 GPU 的显存使用、进程状态等：

pip install nvitop

然后运行命令：

nvitop

这样你可以实时查看显存占用的变化情况，包括每个进程占用的显存量。

方法 2：使用 nvidia-smi 查看 GPU 显存使用

虽说也是使用nvidia-smi，但是不用一直手动点击，只需一次即可。

nvidia-smi 是 NVIDIA 显卡自带的管理工具，可以查看 GPU 的显存、温度、功耗等信息。在命令行中输入以下命令即可：

watch -n 1 nvidia-smi

• watch -n 1：每 1 秒刷新一次显示结果。
• nvidia-smi：显示 GPU