Python 谷歌云虚拟机：Cuda输出内存 我最近在谷歌云VM机上建立了一个账户，我使用英伟达特斯拉K-80GPU。 当我运行Pytorch代码时，出现以下错误：

但是当我在Google Colab上运行相同的代码时，它运行得很好。

有一个关于这个问题的非常详细的问题描述。您的Colab环境的可用内存大小是否相同？我建议（其中一项）进行以下代码更改，以减少存储在内存中的数据量：

• 减少笔记本中的批量大小
• 减少笔记本中的输入维度大小

---

由于这看起来CUDA内存不足，因此我们可以从这个答案中看出，缩小批处理可能是一个好方法。此外，您还可以尝试为虚拟机多提供1个GPU。正如我在本文档中看到的：（Control+f表示“K80”）：12GB的GPU内存对应于K80的1个GPU计数，也许您应该在VM上尝试2GPUs，也许您的工作负载需要更多资源。