Linux 下深度學習工具 caffe 安裝

在一臺系統環境較好的 Linux 機器上可以很容易的安裝 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 官網下 http://www.openblas.net/ 下載安裝安裝包,安裝 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。


轉載:http://blog.csdn.net/xuanyuansen/article/details/46786597

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