1. 程式人生 > 實用技巧 >Ubuntu16.04(CPU)安裝caffe

Ubuntu16.04(CPU)安裝caffe

一、判斷是否支援GPU

  輸入命令lspci | grep -i nvidia,回車後如果沒有任何顯示內容,說明不支援GPU。

二、更改下載源

  備份原來的下載源,cp /etc/apt/sources.list /etc/apt/sources_init.list。使用gedit /etc/apt/sources.list 或者 vim /etc/apt/sources.list 開啟sources.list ,將裡面內容替換為如下:

 1 deb http://mirrors.aliyun.com/ubuntu/ xenial main
 2 deb-src http://mirrors.aliyun.com/ubuntu/ xenial main
 3 deb http://mirrors.aliyun.com/ubuntu/ xenial-updates main
 4 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates main
 5 deb http://mirrors.aliyun.com/ubuntu/ xenial universe
 6 deb-src http://mirrors.aliyun.com/ubuntu/ xenial universe
 7 deb http://mirrors.aliyun.com/ubuntu/ xenial-updates universe
 8 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-updates universe
 9 deb http://mirrors.aliyun.com/ubuntu/ xenial-security main
10 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security main
11 deb http://mirrors.aliyun.com/ubuntu/ xenial-security universe
12 deb-src http://mirrors.aliyun.com/ubuntu/ xenial-security universe

  更新工具源,輸入命令: apt-get update ,回車。

  更新工具軟體:輸入命令: apt-get upgrade ,回車。

  執行完此條命令之後會出現某些包無法下載,根據提示執行命令: apt-get update --fix-missing ,然後重新執行一下上條命令: apt-get upgrade 。

三、安裝依賴包

  按照下面的依賴包安裝,每個命令前面加上sudo獲取許可權。

 1 apt-get install libprotobuf-dev 
 2 apt-get install libleveldb-dev
 3 apt-get install libsnappy-dev 
 4 apt-get install libopencv-dev
 5 apt-get install libhdf5-serial-dev
 6 apt-get install protobuf-compiler
 7 apt-get install libgflags-dev
 8 apt-get install libgoogle-glog-dev
 9 apt-get install liblmdb-dev
10 apt-get install libatlas-base-dev

四、安裝git 以及用git clone指令拉取遠端倉庫GitHub上的caffe專案

  分別輸入命令apt-get install git 、 git clone git://github.com/BVLC/caffe.git。

下載完成後,cd caffe進入caffe目錄,執行命令:cp Makefile.config.example Makefile.config 來備份Makefile.config.example檔案。

  修改新檔案Makefile.config中的配置,鍵入命令: gedit Makefile.config 或者 vim Makefile.config 。

  ①、去掉CPU_ONLY前面的#號註釋,使得CPU_ONLY := 1

  ②、在INCLUDE_DIRS和LIBRARY_DIRS後面分別加入/usr/include/hdf5/serial 和 /usr/lib/x86_64-linux-gnu/hdf5/serial ,儲存退出。

1 # Whatever else you find you need goes here.
2 INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
3 LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial

  ③、如果是python3環境下使用caffe,需要取消對PYTHON_INCLUDE的註釋(第79行),並對之前的python2對應的PYTHON_INCLUDE進行註釋。

  ④、分別輸入命令sudo make all、sudo make test和sudo make runtest,編譯並測試。如果出現錯誤,輸入sudo make clean,解決錯誤後重新編譯和測試。

五、編譯python3介面

  ①、先安裝python3.5,命令為sudo apt-get install python3-pip。

  ②、安裝Python介面依賴庫,在caffe根目錄下,有個python資料夾,資料夾裡面有個requirements.txt,裡面有需要的依賴庫和版本資訊,按照其安裝即可,在安裝前,需要先安裝fortran編輯器(gfrotran),因為安裝scipy庫時需要它,命令

1 sudo apt-get install gfortran
2 cd ~/caffe/python
3 for req in $(cat requirements.txt); do pip install $req; done

  安裝結束後,可以執行如下語句驗證:

1 sudo pip install -r requirements.txt

  ③、將caffe根目錄下的python資料夾加入到環境變數

    開啟配置檔案bashrc,輸入sudo gedit ~/.bashrc命令,在最後一行加入export PYTHONPATH=/home/cff/caffe/python:$PYTHONPATH語句。

    執行更新配置,命令為sudo ldconfig

  ④、編譯python介面

    輸入命令cd ~/caffe/ ,執行make pycaffe

  ⑤、驗證python介面

    進入python3環境,匯入import caffe,如果沒有出錯,安裝成功。

原文:https://blog.csdn.net/u010193446/article/details/53259294