examples/mnist/lenet_solver.prototxt 中
#solver mode:CPU or GPU
solver_mode:GPU
改为
#solver mode:CPU or GPU
solver_mode:CPU
再运行 examples/mnist/train_lenet.sh 就可以正常训练!
examples/mnist/lenet_solver.prototxt 中
#solver mode:CPU or GPU
solver_mode:GPU
改为
#solver mode:CPU or GPU
solver_mode:CPU
再运行 examples/mnist/train_lenet.sh 就可以正常训练!