如何检查 GPU 活动 Colab

在 Colab 中，我们可以使用以下命令来检查 GPU 的活动情况：

!nvidia-smi

这个命令可以显示 GPU 的基本信息、内存使用情况和 GPU 利用率等。但是这个命令并不能持续监控 GPU 的活动情况。如果我们希望能够持续监控 GPU 的活动情况，可以通过以下步骤实现：

步骤 1：安装 GPU 监视器

我们可以通过以下命令来安装 GPU 监视器：

!pip install gpustat

步骤 2：运行 GPU 监视器

安装完成后，我们可以通过运行以下命令来启动 GPU 监视器：

!watch -n 0.1 gpustat --color

在这个命令中，-n 0.1 表示每隔 0.1 秒更新一次 GPU 监视器的信息，--color 表示启用颜色输出。

步骤 3：停止 GPU 监视器

如果我们想要停止 GPU 监视器，可以直接按 Ctrl+C 终止进程。

通过上述步骤，我们就可以方便地实现 Colab 中 GPU 的持续监控，并及时发现 GPU 的占用情况，从而更好地管理 Colab 环境。