如何修复这个奇怪的错误:“RuntimeError: CUDA error: out of memory”

新手上路,请多包涵

我成功地训练了网络,但在验证过程中出现了这个错误:

RuntimeError:CUDA 错误:内存不足

原文由 xiaoding chen 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 324
1 个回答

出现此错误是因为您的 GPU 内存不足。

解决它的一种方法是减小批处理大小,直到您的代码运行时没有此错误。

原文由 K. Khanda 发布,翻译遵循 CC BY-SA 4.0 许可协议

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题