小编Har*_*sad的帖子

我正在使用 google colab 免费 Gpu 进行实验，并想知道有多少 GPU 内存可用于播放，torch.cuda.memory_allocated() 返回当前占用的 GPU 内存，但我们如何使用 PyTorch 确定总可用内存。