Caffe | 检查失败:错误== cudaSuccess(2对0)内存不足

Question

我正在尝试在Caffe上训练一个网络.我的图像大小为512x640.批量大小为1.我正在尝试实施FCN-8.

我目前在具有4GB GPU内存的Amazon EC2实例(g2.2xlarge)上运行此操作.但是当我运行求解器时,它会立即抛出错误

Check failed: error == cudaSuccess (2 vs. 0)  out of memory
*** Check failure stack trace: ***
Aborted (core dumped)

有人可以帮助我从这里开始吗？

Answer 1

你得到的错误确实是内存不足,但它不是RAM,而是GPU内存(注意错误来自CUDA).
通常,当caffe内存不足时 - 首先要做的是减少批量大小(以梯度精度为代价),但是因为你已经批量大小= 1 ...
你确定两者的批量大小都是1 火车和测试阶段？