Ubuntu 14.04 64位配置Caffe 教程（基于CUDA 7.5）

Posted 2020-08-06

　　深度学习是研究计算机视觉的重要工具，尤其在图像分类与识别等领域有着划时代的意义。现在有很多深度学习框架，Caffe是比较常用的一个。本文讲述了Ubuntu 14.04（64位）系统下配置Caffe的基本步骤，参考了Caffe的官方网站   http://caffe.berkeleyvision.org/。

 

一、系统环境配置

 

1.1 首先安装一般会用到的一些依赖项。打开Ubuntu系统的终端，输入以下命令：  

　　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

 

1.2 下载并安装CUDA

　　我使用的是CUDA 7.5，下载地址 https://developer.nvidia.com/cuda-downloads

　　选择合适的版本，下载下来即可。按照图中的设置，下载到的是cuda_7.5.18_linux.run，默认下载路径是～/Downloads，在终端中输入： 

　　cd ~/Downloads

　　sudo sh cuda_7.5.18_linux.run 

 

　　运行后会有一大段文字要让你浏览，终端左下角有进度显示，当前应为0%，没必要看，直接按Ctrl+C 跳过，之后要输入“accept”。在下边就是要你选择装各个组件。第一个显卡驱动不要装，剩下的可以都装上。注意，也不要自己另外装NVIDIA显卡的官方驱动，可能会产生系统启动循环登录不进去的情况。我当时就是因为装了最新版官方驱动，导致循环登录不进去，尝试了网络上各路大神给出的方法，均不奏效。最后放弃了，还是使用自带的开源显卡驱动好了。 

　　但是即使这样可能会出现一些特殊情况。比如使用了自带的开源驱动，在编译完成caffe后，运行使用GPU的测试，会提示Error: CUDA driver version is insufficient for CUDA runtime version 。这表示显卡驱动没装对。这时只能不使用GPU，仅仅使用CPU了。当然如果非要使用GPU，只能安装合适的显卡驱动。正如前文所说，安装了最新的NVIDIA显卡驱动反而造成Ubuntu登录出现问题，那么应该装哪个驱动呢？

　　打开System Settings ——> Software & Updates——> Additional Drivers

 

　　我的显卡是GTX 760，列表中显示驱动中较新的是352.63版本。于是，我们可以从NVIDIA官网下载这个版本的显卡驱动。根据自己的显卡和系统来下载对应的驱动，我的下载后是NVIDIA-Linux-x86_64-352.63.run

　　安装方法：

　　（1）先关闭桌面系统，终端输入：

　　 sudo service lightdm stop

　　（2）然后按Ctrl + Alt +F1进入控制台，输入帐号和密码。cd到显卡驱动所在目录，然后执行：

 　　sudo  sh NVIDIA-Linux-x86_64-352.63.run

　　按照指令一步步安装完即可。

　　（3）中间可能会要求禁用自带开源驱动并重启计算机。重启后需要重复之前的步骤，然后一路确认就可以了。安装完成后，输入命令：

　　 sudo service lightdm start

　　即可返回桌面系统。

 

1.3 安装BLAS

　　终端中输入以下命令： 

　　sudo apt-get install libatlas-base-dev

 

1.4 安装Python（可选的）

　　如果需要使用内置的Python，就需要运行下面的命令，这样在生成pycaffe接口时候才能有Python头文件：

　　sudo apt-get install the python-dev 

 

1.5 剩下的一些依赖 

　　sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

 

二、Caffe的安装和编译

 

2.1安装Caffe

　　我们使用git来获取Caffe，如果之前没有安装git，需要手动安装

　　sudo apt-get install git 

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

 

2.2 配置Caffe 

　　进入caffe的根目录。我的caffe下载后在主目录下，所以在终端输入：

　　cd ~/caffe

 

　　然后创建一个配置文件。由于下载后的caffe目录中自带一个Makefile.config.example文件，所以只需要将该文件复制一份，并且命名为Makefile.config即可。所以可以使用下面的命令：  

　　cp Makefile.config.example Makefile.config

 

　　关于Makefile.config的一些说明如下：

　　（1）使用CPU&GPU加速的Caffe，不需要修改现有配置；

　　（2）使用NVIDIA的cuDNN软件加速的Caffe，则需要取消 “USE_CUDNN := 1”之前的注释符号。需要指出的是，cuDNN并不是一定会比Caffe的GPU加速快。

　　（3）仅使用CPU的Caffe，取消 “CPU_ONLY := 1”前面的注释。

 

2.3 编译Caffe

　　我们使用CMake来编译Caffe，如果之前没有安装CMake，需要手动安装 

　　sudo apt-get install cmake

 

　　然后一次执行以下命令(如果遇到执行权限问题，在命令前加上 sudo) 

　　mkdir build

　　cd build

　　cmake ..

　　make all

　　make install

　　make runtest

 

　　至此，已经成功安装了Caffe。可以在一些数据集上进行一些测试了。

 

三、使用Caffe

 

3.1 准备MNIST 数据集

　　首先准备数据集。我们需要从MNIST网站上下载数据并转换格式。为了简单，可以直接执行下面的命令: 

　　cd ～/caffe

　　./data/mnist/get_mnist.sh

　　./examples/mnist/create_mnist.sh

 

　　运行后，在caffe/examples/mnist下会生成mnist_train_lmdb和 mnist_test_lmdb两个文件夹。

 

3.2 执行结果

　　终端输入以下命令（当前工作目录仍然是～/caffe）: 

　　./examples/mnist/train_lenet.sh

 

　　运行这句命令，最后训练成功结果如图。