ubuntu 安装Nvidia驱动、CUDA、CUDNN

ubuntu 安装Nvidia驱动/CUDA/CUDNN 保姆级教程

该教程目前测试ubuntu18和20都成功了
显卡：GTX 1060 6G

一、安装NVIDIA驱动

1，禁用ubuntu自带的nouveau驱动

在终端中（直接打开的，不要在其他目录中），进行如下输入，第一次打开后需要等待时间较长时间（3min）：

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

在打开的文件末尾加入（有的教程仅仅添加几个然后报错了再补，笔者认为一次性添加上最好）：

blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

编辑保存。执行如下指令关闭nouveau：

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

完成后会显示"options nouveau modeset=0"，更新配置：

sudo update-initramfs -u

完成后会显示"update-initramfs: Generating /boot/initrd.img-5.15.0-83-generic"

上面的禁用效果，必须在重启后采用检验是否成功（一般屏幕显示分辨率会明显下降），可以在终端输入如下命令进行重启：

sudo reboot

重启成功后打开终端输入如下，如果什么都不显示，则正面上面禁用nouveau的流程是正确的

lsmod | grep nouveau

2.获取Kernel source；辅助依赖安装

获取Kernel source：

sudo apt-get install linux-headers-$(uname -r)

这一步最好添加上，防止后面报错；安装依赖，如果这部分没有安装，后续驱动和cuda可能报错

查看是否安装了gcc（一般刚重装完是没有这个的）

gcc -v

输入下面的命令，直接把包括gcc在内很多开发工具包一同安装

sudo apt-get update
sudo apt-get install gcc
sudo apt-get install build-essential
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install pkg-config

安装kmod（不然后面挂载modprobe会报错）

sudo apt-get install kmod

同时需要参见如下流程配置kmod的环境变量，不然后续可能无法调用kmod，利用如下命令打开环境变量(第一次会很慢)：

sudo gedit ~/.bashrc

需要在文件末尾添加的环境变量如下：

export PATH=$PATH:/sbin:/usr/sbin:/usr/local/sbin

保存退出，更新环境变量

source ~/.bashrc

3.根据GPU型号指定版本的驱动

在[英伟达][https://www.nvidia.cn/Download/index.aspx?lang=cn]网址中下载对应显卡型号的驱动程序

4.卸载原有驱动

卸载原有驱动（如果没有可以不操作，如果后面安装选项选错了，也可以选择这个）,这个命令目前看是比较稳定的。如果正确卸载会出现卸载过程。

sudo apt-get remove nvidia-*

5.安装驱动

添加驱动程序的权限：

sudo chmod  a+x NVIDIA-Linux-x86_64-535.104.05.run

利用如下指令进行安装：

sudo ./NVIDIA-Linux-x86_64-535.104.05.run -no-x-check -no-nouveau-check -no-opengl-files

# 参数解释
–no-x-check：表示安装驱动时不检查X服务（图形接口服务），如果没有关闭图形界面则必须加上，否则反之。
–no-nouveau-check：表示安装驱动时不检查nouveau驱动，这也是非必需的，因为我们已经在前面步骤中禁用驱动。
–no-opengl-files：表示只安装驱动文件，不安装OpenGL文件。这个参数不可省略，否则会导致登陆界面死循环，英语一般称为”login loop”或者”stuck in login”。 

安装过程选择：

continue installation

The distribution-provided pre-install script failed! Are you sure you want to continue? 
选择 yes 继续。

Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 
选择 No 继续。

选择install without signing

Nvidia's 32-bit compatibility libraries? 
选择 No 继续。

Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.  
选择 Yes

驱动安装完毕，挂载驱动（有时没有这个也可以）

modprobe nvidia

6.检测是否Nidia显卡驱动是否安装成功

采用如下命令进行测试

nvidia-smi

若报错则配置一下环境变量：

sudo vim ~/.bashrc

在文件末尾添加以下行，并保存退出：

bashCopy codeexport PATH=$PATH:/usr/local/nvidia/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/nvidia/lib64

更新环境变量， 即可

source ~/.bashrc

二、安装CUDA

1.下载cuda安装包

去如下官方网站，选择所需的cuda版本:

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

选择runfile(local)版本，目前有deb、network、run三种模式，虽然直接采用deb不乏成功案例，但是目前run鲁棒性更好，本文采用run进行安装，一个典型的下载及安装命令进行下载**（考虑到wget在网速比较差的情况下不是很稳定，建议另找一个笔记本使用迅雷下载cuda，这样速度和稳定性都更好）**：

https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

2.安装

使用如下命令进行安装，注意要替换成自己的版本：

sudo sh cuda_11.8.0_520.61.05_linux.run

过程会等待一段时间才会弹出安庄界面，需要进行如下选项：

第一个界面输入accept
然后在下一个界面上，取消选择driver，然后选择install

然后进入一段时间，没有任何反映的时间，这是正常的，安装完成会有提示。

3.添加环境变量

首先利用如下命令打开环境变量参数

sudo gedit ~/.bashrc

末尾写入如下：

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

模板是这样的，注意替换自己的版本号：

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

保存更新：

source ~/.bashrc

4.测试CUDA Toolkit

利用如下命令进行检测，如果可以输出当前系统信息而不报错，则完成：

nvcc -V

三、安装cudnn

1.下载cudnn

登录英伟达官网https://developer.nvidia.com/zh-cn/cudnn，下载对应的cudnn版本

2.安装

1. 打开终端，并进入您下载的cuDNN .deb文件所在的目录。

2. 运行以下命令以安装cuDNN：

   sudo dpkg -i libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
   

3. 安装完成后，运行以下命令以更新软件包列表：

   sudo apt update
   # 这将确保系统识别并配置cuDNN的安装。