红联Linux门户
Linux帮助

caffe安装验证

发布时间:2015-12-12 15:19:33来源:linux网站作者:独家记忆_雪1314

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

caffe安装验证


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

然后就可执行成功,最后会出现如图所示结果

caffe安装验证

这里迭代次数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