caffe安装指南(Ubuntu13.04 x86)

时间:2021-03-15 17:20:21

Caffe is a framework for convolutionalneural network algorithms, developed with speed in mind. It was created byYangqing Jia, and is in active development by the Berkeley Vision and LearningCenter.

(以下安装都以Ubuntu13.04为例,并且需要确认有NVIDIA系列显卡)

 

一.前置软件安装

 

1. CUDA(5.0, 5.5, or 6.0).

CUDA ToolKit下载地址https://developer.nvidia.com/cuda-downloads

找到系统对应的,比如我的是Ubuntu 13.04 x86,

就是cuda-repo-ubuntu1304_6.0-37_amd64.deb(下文中有和本文件相关的换成自己系统对应的即可)

1)

$ sudo dpkg --add-architecture armhf
$ sudo apt-get update

(如果是Ubuntu 12.04版本:

$ sudo sh -c \
'echo "foreign-architecture armhf">> /etc/dpkg/dpkg.cfg.d/multiarch'
$ sudo apt-get update

2)

$ sudo dpkg -icuda-repo-<distro>_<version>_<architecture>.deb

实际操作中就是

$ sudo dpkg -i cuda-repo-ubuntu1304_6.0-37_amd64.deb

3)

$ sudo apt-get update
4)

$ sudo apt-get install cuda
5)环境设置(这一步比较重要,会影响到后面caffe的安装)

         在.bashrc文件中加入:

$ export PATH=/usr/local/cuda-6.0/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-6.0/lib64:$LD_LIBRARY_PATH
(32位是/usr/local/cuda-6.0/lib)

 

2. BLAS

可以从ATLAS,Intel MKL,OpenBLAS中选择一个

我这里选择的是ATLAS,安装起来也很方便

$ sudo apt-get install libatlas-base-dev

3. glog(googlelogging library

1)

$ wget https://google-glog.googlecode.com/files/glog-0.3.3.tar.gz
由于国内网络的问题,可能连不上Google服务器,可以到网上找别的地方下

2)

$ tar zxvf glog-0.3.3.tar.gz
3)

$ cd ./glog-0.3.3
$ ./configure

4)

$ make
$ make install

(这里如果有问题可以加上sudo)

5)环境设置

同样是在.bashrc文件中加入:

$ export LD_LIBRARY_PATH=/usr/local/lib


4.其他

$ sudo apt-get install libprotobuf-devlibleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev
 

5. Python and/or Matlab(可选)

由于不是必要的,具体安装就不赘述了,如果有问题可以看看官网。

也可以参考我的另一篇文章 安装Caffe的Python wrapper时出现问题的解决方法

 

二. Caffe的安装

1.下载

https://github.com/BVLC/caffe

网页右侧选择Download ZIP,解压后得到caffe-master文件夹

在caffe-master目录中

1)

$ cp Makefile.config.exampleMakefile.config
这里需要根据自己的需要修改Makefile.config文件

我们这里安装的是CUDA 6.0

在文件中找到CUDA_ARCH一段(第10行起),将最下面两行的注释去掉

另外如果前面BLAS部分选择了MKL或者OpenBlas,只要将21行中BLAS:=后面改掉即可。

还有Matlab和Python相关的可以自行阅读文件中的注释。

 

2)编译

$ make all
$ make test
$ make runtest

如果没有意外的话这三步执行完之后caffe就安装好了。