caffe安裝指南

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的安裝

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就安裝好了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章