ubuntu 14.04下安装配置caffe

本文主要介绍在ubuntu14.04下配置和安装caffe的详细步骤，主要参考资料来源为caffe官网。（安装过程中需要联网）

http://caffe.berkeleyvision.org/

caffe下载链接https://github.com/BVLC/caffe/archive/master.zip

1、安装编译依赖环境

1、打开命令行，输入以下命令

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev

2、安装ATLAS，提供更好的GPU性能支持。

sudo apt-get install libatlas-base-dev

3、python：caffe提供了python接口，ubuntu14.04已经预装了python2.7。你需要安装python-dev包

sudo apt-get install the python-dev

4、其他依赖项

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

（对于ubuntu12.04以下的版本，需要额外安装更多的依赖项，具体参考http://caffe.berkeleyvision.org/install_apt.html）

二、安装cuda

1、检查你的电脑是否支持CUDA。

输入如下命令,如果输出的GPU型号是NVIDIA的,并在此http://developer.nvidia.com/cuda­gpus列表内,则支持。

lspci | grep -i nvidia

如果不支持，则考虑略过此步骤，直接进行cpu模式下的编译。

2、安装

CUDA­7.5在Linux下的安装步骤参见官网手册。,其中提及了“.run”、“.deb”等安装模式，在这里我们以deb包的安装为例。

http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb

1、安装依赖工具:需要连网。

# installation of required tools

sudo apt-get install -y gcc g++ gfortran build-essential \

git wget linux-image-generic libopenblas-dev python-dev \

python-pip python-nose python-numpy python-scipy

2、安装CUDA:

# installing CUDA

sudo dpkg -i cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb

sudo apt-get update

sudo apt-get install cuda

3、配置环境变量

      先使用 sudo vi /etc/profile 打开“profile”文件

       在打开的文件的末尾添加如下代码并保存:

    export PATH=/usr/local/cuda-7.0/bin:$PATH

    export LD_LIBRARY_PATH=/usr/local/cuda-7.0/lib64:$LD_LIBRARY_PATH

       从新加载配置文件

   source /etc/profile</span>

4、重新启动

5、安装samples并检查CPU。samples被默认安装在～/目录下:

cuda-install-samples-7.5.sh ~/

cd NVIDIA_CUDA-7.5_Samples/1_Utilities/deviceQuery

make

./deviceQuery

执行完 ./deviceQuery 命令,终端输出如下类似信息,有GPU型号、驱动版本和运行版本、计算能力......诸多信息: 

至此CUDA安装成功。

3、安装cudnn

cudnn从官网下载，需要注册。如果嫌麻烦，这里我把我的资源分享给大家。

http://yun.baidu.com/share/link?shareid=4252183850&uk=2383908239http://yun.baidu.com/share/link?shareid=4252183850&uk=2383908239

解压cudnn安装包

sudo tar -zvxf cudnn-7.0-linux-x64-v4.0-rc.tgz

进入include目录

sudo cp cudnn.h /usr/local/include

进入lib64目录

sudo cp lib* /usr/local/cuda/lib64

 运行加载操作：

sudo ldconfig

4、编译

解压后进入caffe-master目录

cp Makefile.config.example Makefile.config
gedit Makefile.config

将USE_CUDNN=1取消注释（如果你的电脑不支持navida显卡，则选择取消注释 CPU_ONLY := 1）

# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1

# CPU-only switch (uncomment to build without GPU support).
# CPU_ONLY := 1

接下来就简单了

make all
make test
make runtest

编译需要花很长时间，请耐心等待。。。

到此，caffe就算安装完成了，是不是so easy～

5、编译pycaffe 接口

首先安装pip和python-dev （系统默认有python环境的， 不过我们需要的使python-dev）

sudo apt-get install python-dev python-pip

然后执行如下命令安装编译caffe python wrapper 所需要的额外包

sudo pip install -r /path/to/caffe/python/requirements.txt

在执行上述命令时， 会报错导致不能完全安装所有需要的包。 可以按照官方建议安装anaconda包。 在anaconda官网下载.sh文件，执行，最后添加bin目录到环境变量即可

进入python目录执行

make pycaffe
make distribute

在python目录下就可以import caffe 了

如果想在任意路径下导入，可以修改环境变量

5、总结

整个安装过程中，遇到了很多问题，毕竟不是商业软件，要自己编译，需要各种依赖环境，有时候一个版本号的错误也会导致编译失败。

大家自己安装的时候可以多参考官网，那才是最权威的文档。

遇到问题可以多用google，实在搜不到可以上github和google group 上讨论，上面还是有很多热心的大神的。