【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/