测试是否安装成功caffe

运行mnist

1.下载mnist的数据

#cd caffe

#sudo sh data/mninst/get_mnist.sh

#sudo sh examples/mnist/create_mnist.sh


2.没有GPU配置下prototxt文件

#sudo vim examples/mnist/lenet_solver.prototxt

修改最后一行 solver_mode:GPU 为 solver_mode:CPU


3.运行mnist

#sudo sh examples/mnist/train_lenet.sh


看下运行结果

猜你喜欢

转载自blog.csdn.net/sk_U238/article/details/52382144