Check failed: error == cudaSuccess (2 vs. 0) out of memory 错误。

小编近期在使用caffe进行训练时，常出现  Check failed: error == cudaSuccess (2 vs. 0) out of memory  错误。

从错误提示可以看出是内存不够了。所以小编就加了一个8G 内存条（刚好手边有），可是加上了并没有多大的作用啊，还是照样报错。后来才找到解决方法。

解决方法：

batch_size太大了，一次性读入的图片太多了，所以就超出了显存。因此需要将train.prototxt中的文件train和test的batch_size调小一点。