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

新手上路,请多包涵

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

RuntimeError:CUDA 错误:内存不足

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

阅读 326
1 个回答

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

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

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

推荐问题