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

Display GPU Usage While Code is Running in Colab(代码在Colab中运行时显示GPU使用情况)

本文介绍了代码在Colab中运行时显示GPU使用情况的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

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

推荐答案

使用wandb记录系统指标:

!pip install wandb
import wandb
wandb.init()

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

这篇关于代码在Colab中运行时显示GPU使用情况的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:代码在Colab中运行时显示GPU使用情况

基础教程推荐