1. 程式人生 > >【Caffe】ubuntu14.04下CPU版caffe配置小結

【Caffe】ubuntu14.04下CPU版caffe配置小結


前段時間在win7、ubuntu14.04下都配置過caffe,結合配置流程,把踩過的坑都記錄一下,以供大家參考。以下主要為ubuntu14.04下CPU版的配置過程,其他配置過程可參考最後的補充帖(均已驗證過)。


我的配置過程:(無GPU版)

1.安裝依賴項:

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler libatlas-base-dev

依賴項說明:

主要用到的dependency:

glog:列印除錯資訊,這個對於調錯很有用。
gflags:命令列引數
gtest:測試框架
protobuf:資料的序列化
boost:一些類似C++11的feature,因為早期cuda不支援c++11
opencv:影象處理函式
leveldb,lmdb:簡單的本地資料庫。
cuda:GPU計算框架
atlas/mkl/eigen:線性代數計算庫

2.下載caffe:

sudo git clone git://github.com/BVLC/caffe.git

3.安裝python依賴庫:

檢視“requirements.txt”檔案,用pip工具依次安裝python依賴庫。(假設放置caffe的路徑為$ROOT,根據個人情況修改路徑)
注意:如果使用spyder IDE,需要先在乾淨的系統下安裝spyder,再安裝“requirements.txt”中的依賴項,否則可能因為numpy、scipy等版本衝突導致IDE無法開啟。

cd /$ROOT/caffe/python
for req in $(cat requirements.txt); do
sudo pip install $req; done

4.配置OpenCV環境

Caffe作者預設你已經配置好了OpenCV環境,文件裡沒有說這一步。好在有人已經寫好了配置OpenCV的指令碼, https://github.com/jayrambhia/Install-OpenCV ,直接拿來用。

git clone https://github.com/jayrambhia/Install-OpenCV
cd Install-OpenCV/RedHat
sudo ./opencv_latest.sh

5.編譯caffe:

根據需要修改配置檔案:

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

此處沒有使用GPU和cuDNN,所以取消CPU_ONLY := 1前面的註釋即可。
然後開始編譯caffe:

make all
make test
make runtest

可用多執行緒加速:

make all -j4  
make test -j4  
make runtest -j4

編譯python介面:

sudo make pycaffe -j4

編譯MATLAB介面:

sudo make matcaffe -j4

6. 新增環境變數:

sudo gedit ~/.bashrc

在文字最後新增這兩句:

export PYTHONPATH=/$ROOT/caffe/python:$PYTHONPATH
export CAFFE_HOME=/$ROOT/caffe

更新環境變數:

source .bashrc

(注意不要漏掉bashrc前面的點)


補充帖

win7下配置caffe:

https://github.com/Microsoft/caffe

官方提供的caffe配置方法:

http://caffe.berkeleyvision.org/installation.html

最簡潔概要的caffe+DIGITS配置

(含CUDA+cuDNN+DIGITS)
http://blog.csdn.net/enjoyyl/article/details/47397505/