caffe安装已经结束了,选择的是TrainingLeNet on MNIST with Caffe,具体网址是:http://caffe.berkeleyvision.org/gathered/examples/mnist.html。
Caffe的配置,第一步首先要下载数据集,这里可以通过如下命令去实现。
1.下载数据集
cd $CAFFE_ROOT(这里是进入caffe安装的根目录,我的是/home/caffe/caffe-master)
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
2.将数据转化为leveldb格式具体指令如下所示:
这里建议是根据提示进行,然后在caffe的根目录下进行执行,不然可能会出现错误。
cd/home/caffe/caffe-master
sh /examples/mnist/create_mnist.sh
3.训练数据
sh sh /examples/mnist/train_lenet.sh
这里提出一些可能在训练时会出现错误,因为我在这里选择的是cpu模式,因此他会提示没有gpu,默认情况下是gpu执行的,所以需要修改一个地方在这个文件下
lenet_solver.prototxt(/examples/mnist/lenet_solver.prototxt)找到
# solver mode: CPU or GPU
solver_mode: CPU
然后就可执行成功,最后会出现如图所示结果
这里迭代次数10000次。耐心等待!
Ubuntu14.02(64位系统)+caffe环境配置:http://www.linuxdiyf.com/linux/16460.html
Ubuntu15.04+caffe+cuda7.5+cudnn+matlab:http://www.linuxdiyf.com/linux/16258.html
Caffe在Ubuntu 14.04 64bit下的安装:http://www.linuxdiyf.com/linux/13008.html
Caffe+Ubuntu 15.04+CUDA 7.0新手安装配置指南:http://www.linuxdiyf.com/linux/12708.html
Caffe+Ubuntu 14.04 64bit+CUDA 6.5配置说明:http://www.linuxdiyf.com/linux/11368.html