Ubuntu下安装Caffe
- 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
- 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
- 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。
Ubuntu下安装Caffe
安装需要的软件:
1. Ubuntu14.0.4
2. cudarepoubuntu150475local_7.518_amd64
3. caffemaster
安装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源文件解压缩,即caffemaster然后切到那个目录中执行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 生成mnisttrain
lmdb和mnisttrainlmdb格式的数据集
25. 训练mnist:sh examples/mnist/train_lenet.sh
以上为使用MNIST数据集进行测试
以上操作完成即表示Caffe已成功安装。