python - 谷歌云虚拟机：Cuda out Memory

Question

我最近在 Google Cloud VM Machine 上设置了一个帐户，我使用的是 Nvidia Tesla K-80GPU。当我运行 Pytorch 代码时，出现以下错误：

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

标签： pythongoogle-cloud-platformpytorch

Answer 1

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

• 减少笔记本中的批量大小。
• 减少笔记本中的输入尺寸。