我因为pytorch某些函数要用，必须要用cuda10以上，所以要更新驱动，折腾了四天，终于成功了，真是不容易。记录下这烦人的驱动。

首先得看一下你的cuda和显卡驱动适用情况，要安装什么样点驱动 cuda与显卡驱动版本对应

一、驱动卸载

      主要使用一条命令来卸载  

sudo apt-get --purge remove nvidia*

   执行这条命令的时候，如果卸载的时候让你选择yes  or no，切记要选择yes;

判断 是否卸载成功，     nvidia-smi是否有显示，若还是有显示，则未卸载成功，可以试试删除nvidia-smi,执行下面命令。

sudo rm -r /usr/bin/nvidia-smi

 

二、安装驱动三种方式

1、下载驱动run文件手动安装，这种方式一般会成功，如果没成功，则看第二种方式

2、打开setting--software&Updates--Additional Drivers;

选择一个可用点驱动即可;如果没有想要的驱动，则看第三种方法。

 

3、第三方软件库安装

    3.1 查看推荐版本

ubuntu-drivers devices

可以按照推荐的来，也可以想安装什么就安装。 

  3.2  执行命令

a 添加 PPA 软件仓库：sudo add-apt-repository ppa:graphics-drivers/ppa
b 更新软件索引：sudo apt update
c 终端输入：sudo apt install nvidia-340(340为你安装点驱动版本)

 输入nvidia-smi 如果顺利就安装成功了，如果有问题，就看后面我所遇到点bug.

三、所遇到bug

    1 问题：Failed to initialize NVML: Driver/library version mismatch

       解决办法：重启电脑，一般重启电脑可以解决，如果还不行，可参考https://comzyh.com/blog/archives/967/

2、问题：NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running
以下两个命令即可：

sudo apt-get install dkms

sudo dkms install -m nvidia -v 440.44（440.44表示的是驱动版本号）

                利用命令 ll /usr/src/ 可查看下面有一个nvidia-440.44/文件夹，版本号因电脑而异

四 查看驱动版本

查看GPU型号

lspci | grep -i nvidia

查看NVIDIA驱动版本

sudo dpkg --list | grep nvidia-*

或者

cat /proc/driver/nvidia/version