ubuntu16.04配置caffe+cuda8.0+cudnn5.1

系统：ubuntu16.04
cuda版本：8.0
cudnn版本：5.1

Step 1:安装GPU驱动

1.1 首先使用下面这条命令检查你的电脑上有那些显卡，

$ lspci -k | grep -A 2 -i "VGA"

比如我的显示是这样的，

01:00.0 VGA compatible controller: NVIDIA Corporation Device 1b80 (rev a1)
    Subsystem: Gigabyte Technology Co., Ltd Device 3702
    Kernel driver in use: nouveau

1.2 然后查看哪一个驱动是系统推荐安装的，

$ sudo ubuntu-drivers devices

比如我的是这样的，

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001B80sv00001458sd00003702bc03sc00i00
vendor   : NVIDIA Corporation
driver   : xserver-xorg-video-nouveau - distro free builtin
driver   : nvidia-375 - distro non-free recommended

== cpu-microcode.py ==
driver   : intel-microcode - distro non-free

从中可以看出，推荐安装的是nvidia-375版本的显卡驱动

1.3 禁用nouveau

$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

在最后加入两行，

blacklist nouveau 
options nouveau modeset=0 

命令行执行，

$ sudo update-initramfs -u

修改后需要重启系统。

1.4 确认nouveau是否已经被干掉，使用命令，

$ lsmod | grep nouveau

如果没有显示内容，则说明禁用成功。

1.5 安装GPU驱动
按ctrl+alt+F1进入tty1，关闭桌面服务

$ sudo service lightdm stop

添加graphics driver ppa

$ sudo add-apt-repository ppa:graphics-drivers/ppa

第一次运行，可能会出现警告，回车后继续执行下列命令

$ sudo apt-get update
$ sudo apt-get install nvidia-384

重启桌面服务，按ctrl+alt+F7
查看GPU相关信息

$ nvidia-smi

Step 2：安装cuda8.0版本

cuda是nvidia的编程语言平台，所以想使用GPU必须要安装cuda。
2.1 cuda官网下载相应版本的安装程序，https://developer.nvidia.com/cuda-downloads
这里，我选择的是run文件类型