本文介绍了代码在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,您可以在其中查看不同系统指标的各种图表。
这篇关于代码在Colab中运行时显示GPU使用情况的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持跟版网!
本站部分内容来源互联网,如果有图片或者内容侵犯了您的权益,请联系我们,我们会在确认后第一时间进行删除!