Ubuntu 16.04上安裝Caffe(CPU only)

2020-06-16 17:33:57

作業系統: Ubuntu 16.04

是否使用Python API: 是, 目標是安裝後CAFFE能作為PYTHON MODULE來使用

硬體: 低端筆電, 只使用CPU模式

這是我安裝的過程, 並非安裝教學. 相同情況下可以參考, 能省下一些看文件的時間.

參考: CAFFE官網安裝指南頁面http://caffe.berkeleyvision.org/installation.html#compilation

第一部分: 安裝dependencies

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libatlas-base-dev

sudo apt-get install libhdf5-serial-dev

PYTHON需要2.7版本,這是作業系統本身已經安裝好的. 輸入python2.7 --version 會顯示具體的版本號說明安裝了.

但是還需要sudo apt-get install python-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

(以上所有sudo apt-get在什麼資料夾下輸入都無所謂.)

然後把CAFFE的原始碼下載下來: git clone https://github.com/BVLC/caffe.git

(當然沒有安裝GIT的得先安裝一下)

下載完成之後,進入CAFFE資料夾, 進入裡面的PYTHON資料夾,然後輸入

for req in $(cat requirements.txt); do pip install $req; done

(PIP如果沒有安裝得先安裝一下:sudo apt install python-pip)

第二部分: 安裝CAFFE

到CAFFE資料夾, 使用模板寫個Makefile.config. 具體就是先複製一下模板, 再改一些內容(我喜歡用EMACS).

cp Makefile.config.example Makefile.config

-因為CPU MODE, 所以在CPU_ONLY := 1前面的#要去掉.

-兩個路徑要改成這樣:(新增後面的兩個hdf5的路徑, 否則編譯時報hdf5錯誤)

# Whatever else you find you need goes here.

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

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

準備好了.

make pycaffe

make all

make test

make runtest

--結果顯示ALL TESTS PASSED就安裝好了, 只需要再加上一個PYTHONPATH;

另外, 這個make預設是用CPU單核運算,如果想要快一點, 比如我想使用四核, 在make後面加上-j4標籤.

如果上面4行某一行報錯之後想要重試,建議先make clean再重新開始.

第三部分: 設定 Python Caffe

去到CAFFE資料夾裡面的python資料夾, 把當前路徑記錄下來(pwd). 然後輸入以下命令(把記下的路徑放在相應地方)

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

這時候應該可以了,試驗一下:

$ python2.7

Python 2.7.12 (default, Jul 1 2016, 15:12:24)

[GCC 5.4.0 20160609] on linux2

Type "help", "copyright", "credits" or "license" for more information.

>>> import caffe

>>>

說明安裝全部完成!

第四部分: 錯誤

基本上所有錯誤都是因為dependencies缺乏或者路徑不對,所以根據具體的錯誤資訊對症下藥.例如:

-編譯時顯示hdf5錯誤, 按照上面所說, 增加路徑之後就解決了.

-import caffe時顯示scikit-image錯誤, 那就安裝一下scikit-image就好了.

pip install scikit-image

很多人都在安裝CAFFE時遇到各種錯誤, 請給多一點點耐心, 多谷歌一下, 一定能完成的。

Ubuntu 16.04下Matlab2014a+Anaconda2+OpenCV3.1+Caffe安裝 http://www.linuxidc.com/Linux/2016-07/132860.htm

Ubuntu 16.04系統下CUDA7.5設定Caffe教學 http://www.linuxidc.com/Linux/2016-07/132859.htm

Caffe在Ubuntu 14.04 64bit 下的安裝 http://www.linuxidc.com/Linux/2015-07/120449.htm

深度學習框架Caffe在Ubuntu下編譯安裝 http://www.linuxidc.com/Linux/2016-07/133225.htm

Caffe + Ubuntu 14.04 64bit + CUDA 6.5 設定說明 http://www.linuxidc.com/Linux/2015-04/116444.htm

Ubuntu 16.04上安裝Caffe http://www.linuxidc.com/Linux/2016-08/134585.htm

Caffe設定簡明教學 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 ) http://www.linuxidc.com/Linux/2016-09/135016.htm

Ubuntu 16.04上安裝Caffe(CPU only)

熱門文章