Linux Server GPU(Ubuntu 15.04)安装caffe

亲测可用

https://peter0749.gitbooks.io/caffe/content/chapter1.html

使用MNIST数据集进行测试,默认是GPU模式,可以通过更改mnist文件下的lenet_solver.prototxt转为CPU模式。copy自http://blog.csdn.net/shiorioxy/article/details/52652831

Caffe默认情况会安装在 Caffe-master,就是解压到那个目录,例如:$ home/username/caffe-master,所以下面的工作,默认已经切换到了该工作目录。下面的工作主要是,用于测试Caffe是否工作正常,不做详细评估。

1.数据预处理

sh data/mnist/get_mnist.sh

2.重建lmdb文件。

Caffe支持多种数据格式输入网络,包括Image(.jpg, .png等),leveldb,lmdb,HDF5等,根据自己需要选择不同输入吧。

sh examples/mnist/create_mnist.sh

生成mnist-train-lmdb 和 mnist-train-lmdb文件夹,这里包含了lmdb格式的数据集

3.训练mnist

sh examples/mnist/train_lenet.sh

至此,Caffe安装的所有步骤完结。

你可能感兴趣的:(Linux Server GPU(Ubuntu 15.04)安装caffe)