深度学习工具caffe具体安装指南

时间:2024-02-16 12:35:58

caffe安装指南—吐血整理

前言:

在一台系统环境较好的linux机器上能够非常easy的安装caffe,可是假设系统本身非常旧,又没有GPU的话。安装就太麻烦了,全部都得从头做起,本文档旨在尽可能覆盖安装所要採的坑。


步骤:

一、caffe是主要是C/C++和python编写的。

首先,得须要将gcc,g++安装好,通过yum安装就能够。


二、cuda的安装,假设机器上配置了NVIDIA系列GPU,则须要安装该驱动,推荐安装该驱动至7.0版本号。推荐同一时候安装cuDNN。

安装能够參考http://blog.csdn.net/xuanyuansen/article/details/43152311。centos系统下类似。


三、caffe的layer开发以google的protobuf为格式,须要安装该依赖,推荐安装2.6.1版本号(gcc版本号推荐4.4.7),下载地址为https://github.com/google/protobuf/releases/download/v2.6.1/protobuf-2.6.1.tar.gz。依照README文件中的步骤安装就可以。


四、python安装,caffe提供了python的接口。能够用python进行相关开发,并且能够在ipython-notebook里面画出网络结构。很方便。为了使用这些功能,推荐在目标机器上编译安装python2.7.10,源代码下载地址见python官网,为了多用户环境考虑,强烈建议安装目标用户自己的python。能够在自己用户的home下建立usr目录,解压源代码后加例如以下配置命令:


./configure --enable-shared \

           --prefix=${HOME}/usr/local \

            LDFLAGS=“-Wl,—rpath=${HOME}/usr/local/lib”

make && make install


即须要python的lib文件能够被其它程序共享,rpath后面就是自已的路径。


安装好后须要配置目标用户下的bashrc文件。增加例如以下行:

PATH=/home/lianhua/usr/local/bin:$PATH

然后source bashrc文件。假设进入python版本号正确则成功安装。否则检查路径又一次编译。


python安装好后须要安装pip。在官网上下载get-pip.py运行就可以,然后安装对应版本号的protobuf,同一时候安装caffe的官方说明安装其它的依赖:

for req in $(cat requirements.txt); do pip install $req; done


五、安装openblas

从openblas官网下下载安装安装包。安装README文件安装就好。make&&make install。可能须要sudo权限。


六、yum依赖安装

參加官方文档http://caffe.berkeleyvision.org/install_yum.html,须要安装上述全部依赖,首先须要安装和系统对应的最新的repository文件,然后运行sudo yum install epel-release,更新网站从而安装最最新的依赖。


七、opencv安装

前置须要安装较新版本号的cmake,去官网下载安装就可以。

opencv安装有两种方式:

1、下载源代码包安装,请參考:

http://www.javieriparraguirre.net/installing-opencv-debian/(依赖的安装。centos类似)

http://www.jianshu.com/p/a41da779f763(opencv安装)

2、利用自己主动脚本安装:

地址:https://github.com/jayrambhia/Install-OpenCV



八、caffe安装配置

到此处最终能够进入正题,进行caffe的安装,假设上述安装过程顺利的话,caffe的安装也就没有太大问题了。

关键在于正确配置Makefile.config文件。须要改动例如以下配置至对应的路径(參照配置文件的凝视,疑惑的地方google就可以):

1、CUDA_DIR

2、BLAS := open,BLAS_INCLUDE :=。BLAS_LIB :

3、PYTHON_INCLUDE :=

注意此处须要加入python下numpy的路径

4、PYTHON_LIB :=

5、WITH_PYTHON_LAYER := 1

注意此处须要将python的hdf5安装路径加入到:INCLUDE_DIRS := $(PYTHON_INCLUDE)和LIBRARY_DIRS := $(PYTHON_LIB)


上述配置文件改动好后。回到caffe的根文件夹。运行make all -j4,注意此处进程不要过多,否则会出现诡异的错误。

make test -j4

make runtest -j4

至此caffe主程序编译完成。

以下编译pycaffe,至运行

make pycaffe

make distribute

运行完后改动bashrc文件,加入

PYTHONPATH=${HOME}/caffe/distribute/python:$PYTHONPATH

LD_LIBRARY_PATH=${HOME}/caffe/build/lib:$LD_LIBRARY_PATH

使得python可以找到caffe的依赖。

进入python,import caffe。假设成功则说明一切ok,否则检查路径从头再来,甚至须要又一次编译python。


PS:

出问题可一直google,bless。。!