Caffe | Check failed: error == cudaSuccess (2 vs. 0) out of memory

Question

I am trying to train a network on Caffe. I have image size of 512x640. Batch size is 1. I\'m trying to implement FCN-8s.

I am currently running this on a Amazon EC2

Answer 1

The error you get is indeed out of memory, but it's not the RAM, but rather GPU memory (note that the error comes from CUDA).
Usually, when caffe is out of memory - the first thing to do is reduce the batch size (at the cost of gradient accuracy), but since you are already at batch size = 1...
Are you sure batch size is 1 for both TRAIN and TEST phases?