21天实战caffe的第一个训练程序求助

由于是笔记本上的虚拟机安装的caffe,caffe的编译选项也选择了CPU-only,在运行mnist脚本时遇到如下问题:
caffe>> ./examples/mnist/train_lenet.shCannot use GPU in CPU-only Caffe: check mode.
 
请问该如何破解呢?这个程序必须运行在有GPU的环境嘛?谢谢!
已邀请:

yulizzz

赞同来自:

把examples/mnist/lenet_solver.prototxt里面最后一行solver_mode改成CPU试试

chinaleo - Be quiet

赞同来自:

谢谢,work了!

要回复问题请先登录注册