參考文章https://blog.csdn.net/xuzhongxiong/article/details/52717285
安裝依賴庫
sudo add-apt-repository universe
sudo apt-get update -y
sudo apt-get install cmake -y
# General Dependencies
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev \
libhdf5-serial-dev protobuf-compiler -y
sudo apt-get install --no-install-recommends libboost-all-dev -y
# BLAS
sudo apt-get install libatlas-base-dev -y
# Remaining Dependencies
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev -y
sudo apt-get install python-dev python-numpy –y
sudo apt-get install -y python-pip
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy
編譯 Caffe,cd到要安裝caffe的位置
(1)將終端cd到要安裝caffe的位置桑嘶。
(2)從github上獲取caffe:
git clone https://github.com/BVLC/caffe.git
注意:若沒有安裝Git,需要先安裝Git:
sudo apt-get install git
(3)因為make指令只能make
Makefile.config文件,而Makefile.config.example是caffe給出的makefile例子,因此民宿,首先將Makefile.config.example的內(nèi)容復(fù)制到Makefile.config:
sudo cp Makefile.config.example Makefile.config
(4)打開并修改配置文件:
sudogedit Makefile.config#打開Makefile.config文件
根據(jù)個人情況修改文件:
a.若使用cudnn多律,則
將#USE_CUDNN := 1修改成: USE_CUDNN :=1
b.若使用的opencv版本是3的,則
將#OPENCV_VERSION := 3 修改為: OPENCV_VERSION :=3
c.若要使用python來編寫layer摔认,則
將#WITH_PYTHON_LAYER := 1 修改為 WITH_PYTHON_LAYER :=1
d.重要的一項 :
將# Whatever else you find you need goes here.下面的
INCLUDE_DIRS:=$(PYTHON_INCLUDE) /usr/local/includeLIBRARY_DIRS:=$(PYTHON_LIB) /usr/local/lib /usr/lib 修改為: INCLUDE_DIRS:=$(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS:=$(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/usr/lib/x86_64-linux-gnu/hdf5/serial
這是因為ubuntu16.04的文件包含位置發(fā)生了變化,尤其是需要用到的hdf5的位置绸栅,所以需要更改這一路徑.
(5)修改makefile文件
打開makefile文件级野,做如下修改:
將:NVCCFLAGS+=-ccbin=$(CXX)-Xcompiler-fPIC$(COMMON_FLAGS)替換為:NVCCFLAGS+=-D_FORCE_INLINES-ccbin=$(CXX)-Xcompiler-fPIC$(COMMON_FLAGS)
(6)編輯/usr/local/cuda/include/host_config.h
將其中的第115行注釋掉:
將#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!改為//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
(7)#注意自己CUDA的版本號!
sudocp/usr/local/cuda-8.0/lib64/libcudart.so.8.0/usr/local/lib/libcudart.so.8.0&& sudo ldconfig
sudocp/usr/local/cuda-8.0/lib64/libcublas.so.8.0/usr/local/lib/libcublas.so.8.0&& sudo ldconfig
sudocp/usr/local/cuda-8.0/lib64/libcurand.so.8.0/usr/local/lib/libcurand.so.8.0&& sudo ldconfig
(7)編譯
sudo make all -j8根據(jù)自己電腦配置決定
sudo make runtest
sudo make pycaffe