Ubuntu16.04安装CUDA9.0+CUDNN

写在前面：如果是新装的Ubuntu系统，请先换源。可以考虑清华源：
将/etc/apt/sources.list中内容改为

# 默认注释了源码镜像以提高 apt update 速度，如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse

# 预发布软件源，不建议启用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

安装CUDA

安装CUDA是坑最多的，也是最麻烦的。

1.需要更换Linux内核，我的当前内核版本为4.15，需要换到4.4

a.查看当前安装的内核版本
dpkg -l|grep linux-image
b.查看可以更新的内核版本
sudo apt-cache search linux-image
c.安装新内核
sudo apt-get install linux-image-4.4.0-77-generic linux-image-extra-4.4.0-77-generic
d.更新grub引导
sudo update-grub
e.重启按Esc或Shift+Tab选择Ubuntu高级选项，进入4.4的内核
f.卸载不要的内核
sudo apt-get purge linux-image-xx-xx-generic linux-image-extra-xx-xx-generic
g.更新grub引导
sudo update-grub

2.安装缺少的依赖库

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

3.禁用nouveau驱动

a.在/etc/modprobe.d中创建文件blacklist-nouveau.conf
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
很多时候vi不好用，考虑安装vim
b.在文件中输入：

blacklist nouveau
options nouveau modeset=0

c.执行生效
sudo update-initramfs –u
d.查看是否已关闭
lsmod | grep nouveau
若无内容输出，则成功禁止

4.进入命令行界面

重新启动，到登录界面时按Ctrl+Alt+F1

5.关闭图形化界面

sudo service lightdm stop

6.运行下载好的CUDA程序

选择CUDA9.0，此版本兼容性最好，选择runfile安装
sudo sh cuda.run
按Ctril+c跳过协议阅读部分
在安装选项中，不要安装OpenGL，不然后循环停留在登录界面
因为我电脑本没有安装nvidia显卡驱动，装CUDA时顺便装上了
最后显示installed，表示成功了

7.重新启动图形化界面

sudo service lightdm start
按Ctrl+Alt+F7回到登录界面

8.设置环境变量

在~/.bashrc中加入

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

7.验证CUDA安装完成

a.nvcc -V
b.nvidia-smi

安装CuDnn

1.下载对应版本的cudnn

2.解压

tar -zxf cudnn-xxxx.tgz

3.把include/ 和 lib64/目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64里面

sudo cp -P include/cudnn.h /usr/local/cuda-9.0/include
sudo cp -P lib64/libcudnn* /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*

至此，CUDA与CuDnn安装完成。