训练超参数, 出现 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   lydstory  阅读(1275)  评论(0编辑  收藏  举报

编辑推荐:
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示