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包,如果沒有報錯則安裝成功!