1. 程式人生 > >Ubuntu16.04 配置 caffe (僅CPU)

Ubuntu16.04 配置 caffe (僅CPU)

1. 安裝依賴項(如果一次不成功,可多安裝幾次) 



  
  1. 1 sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
  2. 2 sudo apt-get install --no-install-recommends libboost-all-dev
  3. 3 sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
  4. 4 sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev



2.  下載解壓OpenCV3.1


①   OpenCV 下載連結: 

http://opencv.org/releases.html

      下載3.1.0 Sources原始碼:




②   下載的OpenCV原始碼的壓縮包 opencv-3.1.0.zip 的儲存路徑預設是在 home/使用者名稱/下載 下,進入到該目錄。

        解壓並拷貝到home目錄後重命名:


  
  1. cd ~/下載
  2. unzip opencv -3.1
    .0.zip
  3. sudo mv ./opencv -3.1 .0 ~/OpenCV

 ③  進入OpenCV目錄,建立build資料夾:


  
  1. cd ~/opencv
  2. mkdir build
  3. cd build


3. 使用CMake編譯OpenCV



  
  1. sudo apt install cmake
  2. sudo cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
  3. sduo make -j4





如果編譯不成功,可能需要手動下載ippicv_linux_20151201.tgz檔案,之後編譯成功,輸入以下指令完成安裝:

sudo make install
  


4. 驗證OpenCV是否正確安裝


切換到OpenCV下的samples目錄,並編譯samples程式:


  
  1. cd ~/OpenCV/samples
  2. sudo cmake .
  3. sudo make -j $(nproc)



繼續切換到cpp目錄下,執行測試程式


  
  1. cd cpp/
  2. ./cpp-example-facedetect lena.jpg


這是一個人臉檢測的測試程式,執行結果:





5.  下載caffe


caffe原始碼下載地址:  https://github.com/BVLC/caffe


 

將下載的caffe原始碼壓縮包解壓並移動到home目錄下:


  
  1. cd ~/下載
  2. unzip caffe-master.zip
  3. sudo mv ./caffe-master ~/caffe-master


6. 修改Makefile.config配置檔案


複製一份Makefile.config檔案並使用gedit開啟修改配置:


  
  1. cd ~/caffe-master
  2. sudo cp Makefile.config.example Makefile.config
  3. sudo gedit Makefile.config

修改如下節點:

  • #USE_CUDNN:=1 改為 USE_CUDNN:=0   (不使用CUDN,注意去掉前邊的#符號)
  • #OPENCV_VERSION := 3  OPENCV_VERSION := 3 
  • #WITH_PYTHON_LAYER :=1 改為  WITH_PYTHON_LAYER :=1(python介面)


ubuntu16.04版本相比14.04版本,檔案包含位置有變化,所以以下兩處一定要做更改:


INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include 改為:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib  改為:

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial



7. 修改makefile檔案


開啟caffe-master目錄下的makefile檔案,做如下修改:


NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)  修改為:

NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)



8. 編譯caffe


在caffe-master目錄執行終端,輸入:

sudo make all -j4
  




9. 測試caffe是否安裝成功


執行以下指令:

sudo make runtest
  

執行之後有以下結果,說明caffe配置成功: