step1 安装NVIDIA显卡驱动
因为我是重装了系统所以之前没有安过nvidia驱动，如果之前有装得话，可以自行删除，先 通过快捷键Ctrl+Alt+T打开终端
sudo apt-get remove --purge nvidia-*
下面开始安装 首先添加官方源
sudo add-apt-repository ppa:graphics-drivers/ppa
然后刷新软件库并安装(首先先去NVIDIA官网查询自己适合的驱动)
sudo apt-get update
sudo apt-get install nvidia-375 nvidia-settings nvidia-prime
上面的nvidia-375根据你查询的结果自行更改
Ps:这方法适用于Ubuntu16.04其他版本可能有问题 安装完成之后重启电脑 然后在命令行输入
nvidia-smi
出现显卡信息说明安装成功
step2 安裝CUDA8.0
先下载cuda8.0
然后cd进下载目录执行
sudo sh cuda_8.0.61_375.26_linux.run --override
看到这么一步时，选择n
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 *?
(y)es/(n)o/(q)uit: n
step3 修改~/.bashrc
sudo gedit ~/.bashrc
在最后加入
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
step4 安裝cuDNN 6.0
先下载cuDNN 6.0
然后执行
sudo tar xvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
step5 安装依赖及opencv 3.2.0
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential cmake git pkg-config libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev python-pip python-dev python-numpy python-scipy
sudo apt-get install --no-install-recommends libboost-all-dev
pip install --upgrade pip
sudo pip install Cython
cd ~/
wget https://raw.githubusercontent.com/milq/milq/master/scripts/bash/install-opencv.sh
bash install-opencv.sh#这一步我执行了两次才成功
step6 安装Caffe
下载caffe并进入目录
git clone https://github.com/BVLC/caffe.git
cd caffe
复制配置文件
cp Makefile.config.example Makefile.config
修改配置文件
sudo gedit Makefile.config #打开Makefile.config文件 根据个人情况修改文件：
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/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为：
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 /usr/lib/x86_64-linux-gnu/hdf5/serial
这是因为ubuntu16.04的文件包含位置发生了变化，尤其是需要用到的hdf5的位置，所以需要更改这一路径.
step7 编译caffe
make all
make pycaffe
make test
make runtest
在任意位置import caffe
sudo gedit ~/.bashrc
export PYTHONPATH=~/caffe/python:$PYTHONPATH
#上述语句中 “～” 号表示caffe 所在的根目录。
关闭文件，在终端输入下面命令，使环境变量生效
source ~/.bashrc
编译python3版本
移除下面语句前面的注释
# Uncomment to use Python 3 (default is Python 2)
PYTHON_LIBRARIES := boost_python3 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \
/usr/lib/python3.5/dist-packages/numpy/core/include
可能会遇到如下问题
CXX .build_release/src/caffe/proto/caffe.pb.cc
PROTOC (python) src/caffe/proto/caffe.proto
LD -o .build_release/lib/libcaffe.so.1.0.0-rc3
CXX/LD -o python/caffe/_caffe.so python/caffe/_caffe.cpp
/usr/bin/ld: cannot find -lboost_python3
collect2: error: ld returned 1 exit status
make: *** [python/caffe/_caffe.so] 错误 1
这时候，检查是否有如下文件：
!!! 这里的py35根据你的python版本来，我的是python3.5，这里就是py35
ls /usr/lib/x86_64-linux-gnu/libboost_python-py35.so
如果有，说明我们的系统中已经有了这种库文件，只是文件名不同。接下来执行下面语句
cd /usr/lib/x86_64-linux-gnu/
sudo ln -s libboost_python-py35.so libboost_python3.so
重新编译即可。
finish
reference
http://blog.csdn.net/Tang_DH/article/details/52556636
http://bleuren.me/106/install-caffe/