运行时错误：CUDA 内存不足。使用 Yolac 进行训练时

Question

我一直在按照指南尝试使用 PyTorch 为 Yolact 进行训练：https://github.com/dbolya/yolact

当前 GPU 为 RTX2070，使用 11.1.1 的 cudatoolkit。

当我运行以下命令时：

<块引用>

python train.py --config=yolact_base_config --batch_size=8

我一直遇到这个错误

<块引用>

运行时错误：CUDA 内存不足。尝试分配 50.00 MiB（GPU 0； 7.79 GiB 总容量；已分配 5.98 GiB； 22.88 MiB 免费； PyTorch 总共预留了 6.40 GiB)

我已尝试减小 config.py 下 max_size 中图片的最大大小，但 CUDA 错误仍然存​​在。我也减少了批量大小，但似乎没有改善。

Answer 1

这个错误有几个原因，其中一些是

1. 您的模型层数过于密集，因此您的 GPU 无法对其进行训练
2. 图像输入可能太大，GPU 无法处理

可能还有其他一些原因，您可以尝试的是，查看您的配置并减小批量大小，如果您使用的是 8，然后尝试将其减少到 4 或 3 或其他什么，然后看看您的训练在哪里开始< /p>

或

你可以尝试减小输入尺寸，如果图像......或者我可以同时尝试