1. 程式人生 > >Caffe安裝教程:Ubuntu16.04(CPU)

Caffe安裝教程:Ubuntu16.04(CPU)

一、檢查自己電腦是否具有GPU

lspci | grep -i nvidia

如果未顯示任何內容,則證明你的電腦不支援GPU。

二.安裝必要的環境

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 libgflags-dev libgoogle-glog-dev liblmdb-dev  
sudo apt-get install libatlas-base-dev

三.下載caffe

sudo apt-get install git  安裝Git
git clone git://github.com/BVLC/caffe.git  下載caffe,就可以在根目錄下看到有個caffe目錄了

四.編譯caffe

(1)進入caffe目錄

cd caffe/

(2)生成Makefile.config檔案,這裡是將caffe目錄下自帶的Makefile.config.example檔案複製一份並更名為Makefile.config,命令如下:
cp Makefile.config.example Makefile.config

(3)修改Makefile.config檔案中的配置
1  sudo gedit Makefile.config
2  去掉CPU_ONLY前面的#號,使得CPU_ONLY := 1
3  配置引用檔案路徑(主要是HDF5的路徑問題):
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
BLAS := atlas

(4)編譯 Caffe(這個make預設是用CPU單核運算,如果想要快一點, 比如我想使用四核, 在make後面加上-j4標籤。


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

sudo make runtest 

(執行玩runtest後,會有結果顯示:顯示1098個用例執行成功,結果PASSED!,如果執行結果失敗,則需要執行語句sudo make clean,然後解決掉問題重新編譯


sudo make clean #清除原編譯

sudo make -j4  # make -j4或者make -j8 這根據你自己電腦來定

sudo make pycaffe   #最關鍵的一個

sudo  make runtest   #測試編譯)

五.編譯python介面

(1)安裝pip
sudo apt-get install python-pip
sudo apt-get install python-numpy

(2)安裝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

安裝結束後,可以執行如下語句驗證:
sudo pip install -r caffe/python/requirements.txt
(可以看到,安裝成功的,都會顯示Requirement already satisfied,沒安裝成功的,會繼續進行安裝。)

(3)將caffe根目錄下的python資料夾加入到環境變數
先開啟配置檔案bashrc
sudo gedit ~/.bashrc

在檔案的最後面新增
export PYTHONPATH=/home/c/caffe/python:$PYTHONPATH
注意,這裡路徑根據每個人電腦則不一樣

然後執行更新配置(有時更新不起作用時,可以重啟電腦就會有效):
sudo ldconfig

編譯python介面
cd ~/caffe/
make pycaffe


驗證python介面
進行python環境,引入caffe包,如果沒有報錯則安裝成功!