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安装的所有步骤完结。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容