当前位置:文档之家› Ubuntu下安装Caffe

Ubuntu下安装Caffe

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已成功安装。

相关主题