1. 依赖库安装
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
2. 获取源码
git clone https://github.com/BVLC/caffe.git
3.设置Caffe Python接口
cd caffe/python
for req in $(cat requirements.txt); do sudo pip install $req; done
还需要添加环境变量
sudo gedit ~/.bashrc
在文件的最后添加
#caffe python
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
环境变量进行更新
cd
source ~/.bashrc
4.修改配置文件
修改Makefile.config 文件
第一处
USE_CUDNN := 0
改为
#启用cuDNN加速
USE_CUDNN := 0
第二处
CUDA_DIR := /usr/local/cuda
改为
CUDA_DIR := /usr/local/cuda-8.0
第三处
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
改为
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/
5.安装
make all -j
make test -j
make runtest -j
上边的 -j
表示使用所有的CPU线程来进行安装,也可以在j后边添加自己想使用线程的数量。
###5.添加Caffe环境变量
还需要添加环境变量,在终端中输入环境变量
sudo gedit ~/.bashrc
在文件的最后添加
#Caffe
export CAFFE_ROOT=/home/liu/sfw/dlapp/caffe/
环境变量进行更新
cd
source ~/.bashrc
6.安装成功检验
在终端中进入python运行环境
python
当终端进入python环境后,
import caffe
当程序没有报错时,caffe 1 就安装成功了。