该文章安装的是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计算能力稍好一些,若闲麻烦,使用默认即可!