Ubuntu14.04 安装CUDA7.5 + Caffe + cuDNN

本系列文章由 @yhl_leo 出品，转载请注明出处。

文章链接： http://blog.csdn.net/yhl_leo/article/details/50961542

1 版本

• 笔记本：ThinkPad W541
• Ubuntu 14.04(64-bit)
• 双显卡：
  • Intel(R) HD Graphics Family
  • NVIDIA Quadro K2100M
• CUDA 7.5
• cuDNN

2 准备工作

系统原本是想安装版本Wily，之前也一直在使用，但是无奈，电脑无法安装该版本，就选用了14.04。

安装一些基本依赖项：

$ 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
$ sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

安装NVIDIA显卡驱动，我是先卸载已有版本，再安装，卸载方法：

sudo apt-get remove --purdg nvidia*

重启后登录时，可能会循环出现填写登录密码，不能进入系统的情况，按Ctrl+Alt+F1，登录，安装显卡驱动：

$ sudo apt-get install nvidia-current  

或者指定版本:

$ sudo add-apt-repository ppa:xorg-edgers/ppa
$ sudo apt-get update
$ sudo apt-get install nvidia-346

电脑是双显卡，网友建议在安装前禁掉一些服务，以保证成功。我在安装的时候，只禁掉一项：

$ sudo service lightdm stop

（系统启动后，不要登录进入桌面，直接按Ctrl+Alt+F1，如果你玩过云服务器，应该很熟悉，接着输入系统用户名和登录密码，输入上述的指令，回到图形界面的方法是按Ctrl+Alt+F7，记住这个操作，14.04版本出现输入密码不能进入桌面的情况比较常见，我安装搜狗拼音都出现了……）

3 CUDA

网上很多安装6.x，7.0版本的方法非常繁琐，个人选用了相对简单的安装方法。

首先在官网上下载安装文件（链接前文已经提供）：

我下载的是deb本地安装文件，下载完成后，按照文档提示的命令安装：

$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

下载了下方的Installation Guide for Linux，里面有关于环境变量的设置方法：

$ export PATH=/usr/local/cuda-7.5/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

其他安装文件的配置方式，文档里也都有。

4 Caffe

下载caffe：

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

如果你没安装Git，请阅读博客：Ubuntu Git安装与使用。

然后编译caffe：

$ cd ~/caffe
$ sudo cp Makefile.config.example Makefile.config
$ make all

有些教程也配置OpenCV，方法可以阅读博客： Ubuntu 配置OpenCV 。

5 cuDNN

cuDNN是GPU加速计算深层神经网络的库（下载链接，前文已提供）。

本人的下载文件是：cudnn-7.0-linux-x64-v4.0-prod.tgz

在终端中切换到文件所在文件夹，输入下面指令：

$ sudo tar xvf cudnn-7.0-linux-x64-v4.0-prod.tgz
$ cd cuda/include
$ sudo cp *.h /usr/local/include/
$ cd ../lib64
$ sudo cp lib* /usr/local/lib/
$ cd /usr/local/lib
$ sudo chmod +r libcudnn.so.4.0.4
$ sudo ln -sf libcudnn.so.4.0.4 libcudnn.so.4
$ sudo ln -sf libcudnn.so.4 libcudnn.so
$ sudo ldconfig

然后切换到caffe根目录下，将Makefile.config中的USE_CUDNN行前的#去掉：

$ cd ~/caffe
$ sudo vi Makefile.config

保存后重新编译：

$ sudo make clean
$ sudo make all
$ sudo make test
$ sudo make runtest 

本人编译的时候最后一步的时候，出现这样的错误：

libcudart.so.7.5 cannot open shared object file: No such file or directory

网上有一堆介绍，怎么设置环境变量的（我们分明已经设置过了），可以这样检查是否已经添加环境变量：

$ echo $PATH
$ echo $LD_LIBRARY_PATH

可以看到，环境变量已经添加好。

解决方法是这样，将一些文件复制到/usr/local/lib文件夹下：

$ sudo cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcublas.so.7.5 /usr/local/lib/libcublas.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcurand.so.7.5 /usr/local/lib/libcurand.so.7.5 && sudo ldconfig

再次，尝试sudo make runtest命令，出现如下：

至此，caffe安装完成。