在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