Ubuntu14.04下安装Caffe总结

虽然Deep Learning已经不是什么新鲜的东西了，但是由于设备原因，自己一直没有涉足。前几天有幸换了一台自带GPU的Workstation，甚是高兴。于是迫不及待地装了个Ubuntu系统，开始配置DeepLearning框架Caffe。前后共花了差不多两天时间，终于把它配好了。配了这么多年的软件，Caffe应该是我配过最复杂的软件，没有之一。终于知道为什么用Caffe的人这么多，亲自去配它的人这么少了。以下将一步步讲解配置Caffe的过程，希望对初学者有点帮助。PS：虽然官网http://caffe.berkeleyvision.org/installation.html给出了配置的过程，但是非常笼统，看着看着就不知道自己该干嘛了。

 

一、NVIDIA驱动和CUDAToolkit的安装

1.      安装前需要注意的问题：

a.      检查自己的GPU是否支持CUDA，linux可用以下命令来查看GPU型号：

Lspci | grep –I nvidia

只要GPU型号可以在https://developer.nvidia.com/cuda-gpus找到，就是支持CUDA的，比如我的GPU型号是：GeForceGTX Titan

 

b.      检查自己的操作系统是不是CUDA支持的，可以输入以下命令：

uname -m && cat /etc/*release

CUDA支持的操作系统在以下网址可以找到http://docs.nvidia.com/cuda/cuda-toolkit-release-notes/#overview

 

c.      检查操作系统是否安装了gcc，可用以下命令

gcc –version

很多操作系统都预装有gcc，如果你的没有就先安装一下。

 

d.      去https://developer.nvidia.com/cuda-downloads下载最新版的CUDA，目前是6.5的。这里注意选择与自己机器和操作系统相匹配的（比如我的是64位的Ubuntu14.04）。友情提醒：文件有点大，大概1G，网速不好的同学请耐心等待。

 

e.      验证下载的CUDA是否有效：

md5sum 安装包名字>

执行完上述命令后会出现一串md5值，例如我的是:90b1b8f77313600cc294d9271741f4da，看看https://developer.nvidia.com/cuda-downloads/checksums列表中是否能找到你的md5值，如果不能找到，那就说明之前下的安装包有问题，要重新下载。

 

f.       处理版本冲突问题，第一次安装可以跳过这点。如果之前有装过NVIDIA驱动和CUDA的话，要卸载干净，具体卸载方法参见http://developer.download.nvidia.com/compute/cuda/6_5/rel/docs/CUDA_Getting_Started_Linux.pdf

 

 

2.      安装时需要注意的问题：

a.    退出GUI界面，操作方法是：同时按：CTRL+ALT+F1（F2-F6），切换到TTY1-6命令行模式。退出GUI界面后需要输入用户名、密码登陆。登陆后关闭桌面服务：

sudo stop lightdm

这步很重要，直接在GUI界面装肯定是会失败的！

 

b.   官网说安装前需要Interaction with Nouveau，因为Nouveau是一个开源的显卡驱动，Ubuntu14.04默认安装了，但是它可能会影响NVIDIA驱动的安装。这一步我直接跳过了，事实证明并没有影响。(不放心的同学可以按官网http://developer.download.nvidia.com/compute/cuda/6_5/rel/docs/CUDA_Getting_Started_Linux.pdf那样多操作几步。)

 

c.    现在可以开始安装了。先切换到CUDA安装包所在的位置，执行以下命令：

sudo sh cuda_6.5.14_linux_64.run

这边一路accept-yes-enter就行

执行完后出现以下结果：

Driver: Installed requirereboot

Toolkit: installskip

Samples: installskip

看到了吗，这个CUDA安装包事实上包含了Driver，Toolkit和Samples三部分，第一次安装时把驱动装上了，只不过要重新启动。重新启动了一下，还是跟上面一样，退出GUI，重新执行以下命令:

sudo sh cuda_6.5.14_linux_64.run

结果如下：

Driver:Installed

Toolkit: Installed in/usr/local/cuda-6.5

Samples: Installed in/home/Changzhi, but missing recommendedlibraries

这次就安装成功了，但是缺少了一些推荐的库。

 

d.      这一步显然要安装那些推荐的库，没什么好说的，按官网推荐的，需要安装以下库：

sudo apt-get install freeglut3-dev build-essential libx11-devlibxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesalibglu1-mesa-dev

 

 

e.      装完以后就可以回到GUI界面了，执行

sudo start lightdm

 

f.       按照官网http://caffe.berkeleyvision.org/installation.html#prerequisites的要求，还需要安装一些dependencies，我的系统是Ubuntu14.04，所以还需要执行以下两步：

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-devlibopencv-dev libboost-all-dev libhdf5-serial-dev

 

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

 

 

3.      安装后需要注意的问题：

a.      设置环境变量：

在/etc/profile最后面添加一句（添加不了的可以切换到root用户）

export PATH=$PATH:/usr/local/cuda-6.5/bin

保存后执行以下命令，是环境变量立即生效

source /etc/profile

 

 

b.      添加lib库路径

在/etc/ld.so.conf.d加入文件cuda.conf，写入

/usr/local/cuda-6.5/lib64

同样，执行下列命令使其立即生效

sudo ldconfig

 

 

4.      验证安装结果

 

a.      验证驱动的版本，和上述gcc类似，就是验证驱动和有没有问题

cat /proc/driver/nvidia/version

 

 

b.      编译例子

先看一下编译器版本，执行

nvcc –V

          不出意外，这里会提示nvcc没有安装，其实就是之前装的NVIDIA-CUDA-Toolkit的编译器没有安装完整，根据提示安装就好：

               sudoapt-get install nvidia-cuda-toolkit

                    这个安装也需要很长时间，请耐心等待。安装完后就可以编译例子了，例子默认安装在/home/username/NVIDIA_CUDA-6.5_Samples，进入这个目录，执行

               make

        该过程也比较漫长，请耐心等待。

 

c.      Running the binaries.

全部编译玩后，进入/home/username/NVIDIA_CUDA-6.5_Samples/bin/x86_64/linux/release，执行

sudo ./deviceQuery

如果出现以下信息，则驱动及显卡安装成功：

Note： 如果大家第一次安装显示的CUDA driver version和runtimeversion是一样的话，以下标绿的部分可以略过。

我第一次安装时，出现了“cuda driver version isinsufficient for cudaruntime”这个错误，就是默认安装时cuda版本是6.5，但是驱动版本还是5.5。解决这个问题肯定是安装个6.5版的cuda驱动就行了。但问题没有那么简单。我从官网下了6.5的cuda驱动，安装后，别说解决问题，连ubuntu系统都进不去了。开机时就左上角有个小横线在一闪一闪的，还什么都不能输入，真是能把人急死！后来网上找到了解决办法如下：

   login the system using root recover mode,execute:

       sudo apt-get purge nvidia*

       sudo apt-get install nvidia-current

       sudoreboot

 

系统启动不了原因是显卡驱动冲突，所以我们只需卸载干净前一个驱动，只保留一个最新版的驱动即可。这个recovermode开机时按住左边的shift键就可以进入。

 

至此，NVIDIA驱动和CUDAToolkit的安装完成。

 

二、BLAS的安装

这边选择的是很多网友推荐的intel MKL，这个是收费软件，但学生可以申请免费使用，申请地址https://software.intel.com/en-us/intel-education-offerings#pid-2460-93，这里注意选择自己对应操作系统的版本，还要申请时只能用edu的邮箱。这个软件非常大，3G+，下载时可以选择在线安装或者离线下载，这里强烈推荐离线下载再安装，因为我就是在线安装时出现问题了，后来卸载重新离线安装的。我是linux系统，所以下载的是parallel_studio_xe_2015_update1.tgz，解压后进入目录，执行

sudo ./install_GUI.sh

这是GUI界面安装，和windows一样，按照提示一步步安装就行了，路径什么的选择默认就好。装好后需要添加library路径，具体操作是在/etc/ld.so.conf.d/下新建文件intel_mkl.conf，没权限的话切换到root用户来操作，建好后在文件中添加：

/opt/intel/lib/intel64

/opt/intel/mkl/lib/intel64

安装mkl时如果不是选择默认路径的，以上操作记得换成自己的安装路径。添加完内容后执行以下命令使其立即生效。

sudo ldconfig

 

 

三、OpenCV的安装

这个尽量不要手动安装， Github上有人已经写好了完整的安装脚本：https://github.com/jayrambhia/Install-OpenCV，下载后解压，然后进去该目录，选择自己的操作系统，比如我的是Ubuntu，执行

sudo ./dependencies.sh

sudo ./open2_4_9.sh

其中第一步是安装依赖项，第二部是安装opencv，这个过程大概二三十分钟，慢慢等。

 

四、Caffe的安装和编译

Caffe下载地址：https://github.com/BVLC/caffe

下载后，进入Caffe目录执行

cp Makefile.config.example Makefile.config

然后修改Makefile.config中的BLAS:= atlas为BLAS := mkl，保存退出就行了。

最后是编译，执行以下三条命令：

make all

make test

make runtest

前两步是肯定没问题的，第三部runtest，最后会有两项不通过，但是还是pass了，不影响，应该是没装python的原因。如果不需要在python下用caffe，就可以不管了。

至此Caffe安装完成！