Ubuntu安装NVIDIA显卡驱动、cuda、cudnn

一、Ubuntu安装NVIDIA显卡驱动

0.删除旧驱动：

sudo apt-get purge nvidia*

1.将ppa:graphics-drivers/ppa存储库添加到系统中：

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt update

2.识别显卡模型和推荐的驱动程序：

ubuntu-drivers devices

3.输入命令：

sudo apt-get install nvidia-418 nvidia-settings

完成后，重启Ubuntu

4.查看显卡驱动版本：

nvidia-smi

二、Ubuntu安装cuda、cudnn

0.卸载旧cuda

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_10.1.pl

删除cudnn文件：

sudo rm -rf /usr/local/cuda-10.1

1.安装cuda

(1)下载cuda_10.1.243_418.87.00_linux.run：

https://developer.nvidia.com/cuda-toolkit-archive

安装：

sudo sh cuda_10.1.243_418.87.00_linux.run

然后一直按回车，accept ——> n（不要安装driver）——> y ——> y ——> y

(2)安装完成后，设置环境变量

主目录下（/home/sun）打开 .bashrc文件，添加相关路径。如果没有找到，可按Ctrl+H键显示隐藏文件。

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

在终端输入：

source ~/.bashrc

（3）查看当前cuda版本

终端输入：

nvcc --version

3.删除旧cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h

sudo rm -rf /usr/local/cuda/lib64/libcudnn

4.安装新版本的cudnn

(1)下载cudnn-10.1-linux-x64-v7.6.4.38.zip:

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

解压文件，并在解压出的cuda文件夹中打开终端，终端输入：

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

 sudo cp lib64/lib* /usr/local/cuda/lib64/

（2）建立软链接：

终端输入：

cd /usr/local/cuda/lib64

sudo chmod +r libcudnn.so.7.6.4

sudo ln -sf libcudnn.so.7.6.4 libcudnn.so.7  

sudo ln -sf libcudnn.so.7 libcudnn.so 

sudo ldconfig

（3）查看cudnn的版本

终端输入:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果出现上图所示版本信息（三个数字的意思是当前版本为7.6.4），说明安装成功。

参看链接：

https://blog.csdn.net/qq_33591712/article/details/94731966 （安装NVIDIA显卡驱动(超简单)）

https://blog.csdn.net/wanzhen4330/article/details/81704474 （Ubuntu16.04下cuda和cudnn的卸载和升级）