caffe跑试验遇到错误：Check failed: error == cudaSuccess (2 vs. 0) out of memory

caffe跑试验遇到错误：Check failed: error == cudaSuccess (2 vs. 0) out of memory

运行程序，通过下列命令：

6G的1060跑VGG16的SSD跑不动啊！

watch -n 1 nvidia-smi

可以看到显存很快超载

我也碰到了这个问题，我直接设置的batch等于1，因为我学习DIGITS时候发现，就算使用batch等于1，然后训练FCNAlexnet的话，一般gpu的使用率也在peak上，DIGITS本来感觉是个神器，不过等到做自己的数据集的时候（13400张路况图片的时候就无法生成dataset了）。

我看stackflow里面的回答

1.减小mini_batch

2.减小图片大小

3.升级GPU（stackflow里面一般都是8G-12G的，别的CS的实验室好像直接就用的GTX 1080,可以考虑）

作者：冯姆姆
链接：https://www.zhihu.com/question/41700499/answer/182979714
来源：知乎
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

https://www.zhihu.com/question/33505655