Ubuntu安装NVIDIA驱动用于深度学习GPU加速

一、环境

下面是我安装时候的系统环境，其它版本应该也通用。

• ubuntu18.04
• GeForce GTX 1060

二、安装过程

安装方法不止一种，这里介绍其中一种，经过采坑验证过成功的方法。
1.卸载原有NVIDIA驱动（防止之前安装过有冲突）

sudo apt-get remove --purge nvidia*
sudo apt autoremove

2.禁用开源nouveau驱动
首先打开的文本

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

在最后一行加入，拉黑它：

blacklist nouveau

保存退出后执行：

sudo update-initramfs -u

重启，注意这个时候应该是进不去系统了，这时不要慌，进行如下操作：
在重启后第一个界面按一下“e"键，在倒数第二行quiet splash后面输入nomodeset （注意之间有空格隔开的），然后按F10，就可以进入系统了，这时系统界面看起来会特别别扭。
打开终端，检查是否禁用成功：

lsmod | grep nouveau

如果没有屏幕输出，说明禁用nouveau成功

3.使用命令查看哪个版本可以装

ubuntu-drivers devices

4.进入https://www.nvidia.cn/Download/Find.aspx?lang=cn找到对应电脑配置下载NVIDAI驱动（建议不要下载最新版本），比如这里下载的是：NVIDIA-Linux-x86_64-410.78.run
附上我已经下载百度网盘的链接：
链接：https://pan.baidu.com/s/1kwkg4RZOB08Vaxp2OlOjng
提取码：28m7

5.安装

sudo bash NVIDIA-Linux-x86_64-410.78.run

这一步很可能会安装失败
安装过程过会出现蓝色的配置框，注意如果有提示GCC版本不对应的话，一定要先把GCC升级为相应的版本，要不然在后面的编译安装就会失败(我的就是这样的)
如果没提示GCC版本不对应问题，就直接根据默认的配置一直往下按enter就行了
6.之后重启，使用命令查看是否安装成功。

nvidia-smi

三、附加介绍

我通过上面的方法安装失败了无数次，直到把GCC版本升级了，最后才成功的，下面顺便介绍下我是怎么升级的
1.先卸载

sudo apt-get remove gcc
sudo apt autoremove

2.安装

sudo apt-get install gcc
sudo apt-get install build-essential

该方法一般会升级到最新版本，如果想要安装指定版本的，请参考：https://blog.csdn.net/a145127/article/details/88954122