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

Caffe can use multiple GPU's. This is only supported in the C++ interface, not in the python one. You could also enable cuDNN for a lower memory footprint.

https://github.com/BVLC/caffe/blob/master/docs/multigpu.md