在Colab中监测GPU显存占用方法

在Colab中监测GPU显存占用的方法

1.氪金,Colab Pro,开两个Terminal,一个运行,一个监视显存

2.用wandb库

import wandb
wandb.init(project='tensorrt2')

wandb的详细教程请自行百度wandb

https://stackoverflow.com/questions/62620268/display-gpu-usage-while-code-is-running-in-colab

你可能感兴趣的:(在Colab中监测GPU显存占用方法)