训练超参数, 出现 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 就可以正常训练!

转载于:https://www.cnblogs.com/hshy/p/8617751.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值