Ubuntu 16.04 安装NVIDIA显卡驱动及CUDA、CuDnn(详细)

Ubuntu 16.04 安装NVIDIA显卡驱动及CUDA、CuDnn

一、安装NVIDIA显卡驱动

1、首先卸载原有NVIDIA驱动：

sudo apt-get purge nvidia*

2、禁用nouveau驱动

      在终端输入：

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

     添加： 

blacklist nouveau
options nouveau modeset=0

     一定记着更新：

sudo update-initramfs -u   

     更新完成后重启系统

sudo reboot

     重启后输入：

lsmod | grep nouveau

     如果没有输出结果则正确。

3、下载NVIDIA驱动

     http://www.nvidia.cn/Download/index.aspx?lang=cn

选择对应的版本下载

4、在终端输入

sudo service lightdm stop

     按Ctrl+Alt+F1，切换到tty1，需要再次输入Ubuntu用户名和密码。

5、安装驱动

sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files

–no-opengl-files也就是不安装opengl，这个是必须要的，否则可能出现循环登陆的问题，–no-x-check 和–no-nouveau-check 是安装驱动时不检查X服务和nouvea，这两句不是必要的。

6、在终端输入，返回。

sudo service lightdm start

7、检查NVIDIA，终端输入

nvidia-smi

二、安装CUDA

1、进入官网选择对应版本下载：

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

在终端输入：

sudo bash ***.run

注意，此时遇到询问是否安装驱动的选项，选择NO,不再安装驱动

2、添加到环境变量

 （1）终端输入

sudo gedit ~/.bashrc

   添加：

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

 （2）终端输入

sudo gedit /etc/profile

   添加：

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

 （3）终端输入

sudo gedit /etc/ld.so.conf.d/cuda.conf

    添加：

/usr/local/cuda/lib64

  （4）终端输入

sudo ldconfig

3、简单验证：

nvcc -V

三、配置CuDnn

1、官网下载;

https://developer.nvidia.com/rdp/cudnn-download

需要先注册一个账号，再选择对应版本下载

2、解压

3、移动文件

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

4、重新生成软链接（具体内容根据实际情况修改）

sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.0.4 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so

5、完成