代码在Colab中运行时显示GPU使用情况

问题描述

我有一个在Google Colab上运行的程序，我需要在它运行时监视GPU的使用情况。我知道通常您会在命令行中使用nvidia-smi来显示GPU使用情况，但是由于Colab在任何时候只允许一次运行一个单元，所以这不是一个选项。目前我正在使用GPUtil，通过GPUtil.getGPUs()[0].load和GPUtil.getGPUs()[0].memoryUsed监控gpu和vram的使用情况，但是我找不到让这些代码与我的代码的睡觉同时执行的方法，所以使用量远远低于实际应该执行的数量。是否有办法在其他代码运行时打印GPU使用情况？

---

解决方案

使用wandb记录系统指标：

!pip install wandb
import wandb
wandb.init()

它输出一个URL，您可以在其中查看不同系统指标的各种图表。