Caffe训练AlexNet网络模型——问题二

训练时，出现Check failed：error == cudaSuccess (2 vs. 0) out of memory，并且accruary = 0，如下图所示：

解决方法：将train_val.prototxt文件中的batch_size变小一点，如下图所示：

也可参见博客：

http://blog.csdn.net/u013066730/article/details/53784614