Install nvidia-driver ,cuda, cudnn on ubuntu16.04-64bit

基础：

OS：ubuntu16.04-64bit (linux 4.13.0-36)

GPU: gtx-950 

nvidia-driver version :384.111

cuda 9.0

cudnn-9.0-v7.1

相关链接及文件名：

nvidia-driver:  https://www.geforce.cn/drivers   选择合适自己版本的 

cuda 9.0  : https://developer.nvidia.com/cuda-downloads

cudnn-9.0 : https://developer.nvidia.com/cudnn

 

如何判断该安装什么版本的nvidia-driver,cuda,cudnn？

答：在系统中先查看本机的gpu是哪种型号的，可以用lspci |grep VGA 查看，然后用ubuntu-drivers devices 查看出驱动版本，一般在“driver:”那一行会有一个recommended 的版本号，可以去下载。然后选择相应的cuda,最后确定cudnn

依赖关系:driver->cuda->cudnn

安装步骤：

S1.安装好指定系统后，安装64bit 兼容IA32库：

根据该步骤：https://www.91linux.org/89.html 

 

1、首先判断在用的架构

dpkg --print-architecture

输出当前的使用的架构，比如amd64

2、添加架构

dpkg --add-architecture i386

添加i386架构，然后刷新下源列表

sudo apt-get update

这时就已经包含i386的软件列表

 

3、安装i386

apt-get install iceweasel:i386

S2.禁用当前的nouveau 驱动，也即关闭Nouveau：

    1)把驱动加入黑名单中: /etc/modprobe.d/blacklist.conf 在后面加入：

    blacklist nouveau

    2) 使用 dracut重新建立 initramfs image file :

    * 备份 the initramfs file

    $ sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

    * 重新建立 the initramfs file

    $ sudo dracut -v /boot/initramfs-$(uname -r).img $(uname -r)

    3) 重启系统至文本模式,init 3 这个可以修改/etc/inittab 文件 init 3是文本模式。这时候重启就可以看到桌面非常的丑，就说明nouveau已经没有被加载了

    4)检查nouveau driver确保没有被加载！

$ lsmod | grep nouveau

S3.安装nvidia-driver

1.按住CTRL+ALT+F1 进入命令行

2.    sudo service lightdm stop    或者   sudo stop lightdm

3.    sudo init 3

4.  安装驱动程序：     ＃:  sudo    ./NVIDIA-Linux-x86_64-384.111.run      //当前目录下执行NVIDIA驱动程序

5.   按照提示安装完成，简单方法重启就好了     sudo  reboot

S4.安装cuda

sudo ./cuda_9.0176_384.81_linux.run

安装过程中选择询问是否需要安装nvidia-driver驱动时可以选择no ，因为上一步已经安装成功了。

成功后设置环境变量

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

export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64"

运行nvidia-smi可以检查有没有安装成功

S5.安装cudnn

tar xvzf cudnn-9.0-linux-x64-v5.1.tgz

然后将库和头文件copy到cuda目录（一定是你自己安装的目录如/usr/local/cuda-9.0）,

不过正确安装的话，ubuntu一般就会有软链接/usr/local/cuda -> /usr/local/cuda-9.0/

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo

chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*