我正在使用Google Colab免费的GPU进行实验,并想知道可以用于操作的GPU内存有多少。torch.cuda.memory_allocated()返回当前占用的GPU内存,但我们如何使用PyTorch确定总可用内存呢?
用户名或电子邮箱地址
密码