Ubuntu Caffe 安装步骤记录

Ubuntu Caffe 安装步骤记录

安装流程细分为如下10个步骤，细化步骤粒度更易避免出错：

1. 安装依赖包
2. 禁用 nouveau
3. 配置环境变量
4. 下载 CUDA 8.0
5. 安装 CUDA 8.0
6. 验证 CUDA 8.0 是否安装成功
7. 安装 cudnn
8. 安装 opencv
9. 安装 caffe
10. 使用MNIST数据集进行测试

前8步，服务器上已经安装过了，这里只说明第九步

第9步 安装 caffe

首先在你要安装的路径下 clone ：

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

进入 caffe:

cd caffe

将 Makefile.config.example 文件复制一份并更名为 Makefile.config ，也可以在 caffe 目录下直接调用以下命令完成复制操作 ：

cp Makefile.config.example Makefile.config

复制一份的原因是编译 caffe 时需要的是 Makefile.config 文件，而Makefile.config.example 只是caffe 给出的配置文件例子，不能用来编译 caffe。

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

gedit Makefile.config

修改 Makefile.config 文件内容：

1、应用 cudnn

将#USE_CUDNN := 1
修改成： 
USE_CUDNN := 1

我们的服务器上并没有cudnn，这一块 不需要修改，否则会报错

2、根据cuda版本注释掉

#-gencode arch=compute_60,code=sm_60 \
#-gencode arch=compute_61,code=sm_61 \
#-gencode arch=compute_61,code=compute_61

3、应用 opencv 版本

将#OPENCV_VERSION := 3 
修改为： 
OPENCV_VERSION := 2.4

注意：将opencv修改成对应的版本

4、实现caffe对Python和Matlab接口的支持：

PYTHON_LIB := /usr/lib
MATLAB_DIR := /usr/local/MATLAB/R2015b

注意：将Matlab修改成对应的版本

5、编译caffe-master

make all -j8
make test -j8
make runtest -j8
make pycaffe -j8
make matcaffe -j8

编译成功后可运行测试，部分结果如下：

使用MNIST数据集进行测试

Caffe默认情况会安装在 Caffe-master，就是解压到那个目录，例如：$ home/username/caffe-master，所以下面的工作，默认已经切换到了该工作目录。下面的工作主要是，用于测试Caffe是否工作正常，不做详细评估。具体设置请参考官网：http://caffe.berkeleyvision.org/gathered/examples/mnist.html

1、数据预处理

sh data/mnist/get_mnist.sh

2、重建lmdb文件。Caffe支持多种数据格式输入网络，包括Image(.jpg, .png等)，leveldb，lmdb，HDF5等，根据自己需要选择不同输入吧。

sh examples/mnist/create_mnist.sh

生成mnist-train-lmdb 和 mnist-train-lmdb文件夹，这里包含了lmdb格式的数据集

3、训练mnist

sh examples/mnist/train_lenet.sh

至此，Caffe安装的所有步骤完结