美文网首页
Linux Server GPU(Ubuntu 15.04)安装

Linux Server GPU(Ubuntu 15.04)安装

作者: kelseyh | 来源:发表于2017-02-23 19:30 被阅读63次

亲测可用

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)安装

      本文链接:https://www.haomeiwen.com/subject/psnbwttx.html