安装caffe（GPU版）

该文章安装的是GPU版的caffe！

一、安装依赖

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install  -y libopencv-dev
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y liblapack-dev
sudo apt-get install -y libatlas-base-dev 
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install -y python-pip
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy
sudo apt-get install -y python3-dev
sudo apt-get install -y python3-numpy python3-scipy

二、下载caffe

使用一下命令下载

git clone https://github.com/BVLC/caffe.git

三、开始安装

1.进入caffe目录

cd caffe

2.拷贝一个安装配置文件

cp Makefile.config.example Makefile.config

3.然后修改 Makefile.config 文件，在 caffe 目录下打开该文件：

sudo gedit Makefile.config

CPU版
1、将
#OPENCV_VERSION := 3 
修改为： 
OPENCV_VERSION := 3
2.

安装GPU版在以上的基础上还需修改

GPU版
1.应用 cudnn，将
#USE_CUDNN := 1
修改成： 
USE_CUDNN := 1

2.应用 opencv 版本，将
#OPENCV_VERSION := 3 
修改为： 
OPENCV_VERSION := 3

3.修改cuda路径，将
CUDA_DIR := /usr/local/cuda 
修改为 
CUDA_DIR := /usr/local/cuda-9.0

4.修改CUDA_ARCH，将
将
-gencode arch=compute_20,code=sm_20 \
		-gencode arch=compute_20,code=sm_21 \
两行注释或删除

5.修改blas
修改为 
BLAS := mkl
#默认的不是mkl，mkl计算能力稍好一些，若闲麻烦，使用默认即可！