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



你可能感兴趣的:(caffe)