Ubuntu下安装Caffe

合集下载
  1. 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
  2. 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
  3. 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。

Ubuntu下安装Caffe

安装需要的软件:

1. Ubuntu14.0.4

2. cuda­repo­ubuntu1504­7­5­local_7.5­18_amd64

3. caffe­master

安装Ubuntu的过程在这里不做介绍。系统安装好了之后,执行下面的操作。

在线安装的步骤,在此之前将更新源换成163的,下载速度会快很多:

1. sudo apt-get update

2. sudo apt-get upgrade

以上更新系统相关软件

3. sudo apt-get install build-essential

4. sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev

libopencv-dev libboost-all-dev libhdf5-serial-dev

5. sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

protobuf-compiler

6. sudo apt-get install libatlas-base-dev python-dev vim

以上为在线安装的依赖项需要按顺序执行

7. 切换到cuda安装包的目录下,执行sudo dpkg -i cuda-repo-ubuntu1504-7-5-

local_7.5-18_amd64

8. sudo apt-get update

9. sudo apt-get install -y cuda

以上为安装CUDA驱动。(Nvidia驱动在这个过程也会自动装好)

10. vim ~/.bashrc

11. 按i进入编辑状态,将光标移到最下面一行。输入export PATH=/usr/local/cuda-

7.5/bin:$PATH

12. 输入export LD_LIBRARY_PATH=/usr/local/cuda-

7.5/lib64:$LD_LIBRARY_PATH,按:wq保存并退出

13. source ~/.bashrc

以上为配置环境变量

14. 将下载好的caffe源文件解压缩,即caffe­master然后切到那个目录中执行cp

Makefile.config.example Makefile.config

15. vim Makefile.config

16. 按i进入编辑状态,修改:CUDA_DIR :=/usr/local/cuda-7.5

17. 如果使用了MATLAB或者Python需要按照Makefile.config文件中的提示在

Makefile.config进行相应的路径修改即可。

以上为配置Caffe相关路径

18. make all

19. make test

20. make runtest

21. 如果用到MATLAB,make matcaffe

22. 如果用到Python,make pycaffe

以上为编译Caffe

23. 数据预处理:sh data/mnist/get_mnist.sh

24. 重建lmdb/leveldb文件:sh examples/mnist/create_mnist.sh 生成mnist­train­

lmdb和mnist­train­lmdb格式的数据集

25. 训练mnist:sh examples/mnist/train_lenet.sh

以上为使用MNIST数据集进行测试

以上操作完成即表示Caffe已成功安装。

相关文档
最新文档