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