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