训练超参数， 出现 Cannot use GPU in CPU-only Caffe 错误？

当我们用MNIST手写体数字数据库和LeNet CNN 模型训练超参数，运行 examples/mnist/train_lenet.sh是出现Cannot use GPU in CPU-only Caffe: check mode. 

屏幕快照_2017-03-23_下午4.53_.32_.jpg


 
这时候需要调整两个配置文件：
1.安装文件根目录的Makefile.config 里的

屏幕快照_2017-03-23_下午4.56_.56_.jpg


 
这一项去掉#
2. examples/mnist/lenet_solver.prototxt 中

屏幕快照_2017-03-23_下午5.00_.12_.jpg


将这一项de的GPU改为CPU 后
再运行 examples/mnist/train_lenet.sh 就可以正常训练！

posted on 2018-03-21 16:18  lydstory  阅读(1254)  评论(0)  编辑  收藏  举报

导航