Ubuntu20.04安装Nvidia Driver455+Cuda11.1+Cudnn8.0.5

前言

• 博主电脑刚装了Ubuntu20.04，电脑显卡1050Ti，准备用来跑深度学习代码，需要安装Cuda才能支持GPU，刚装成功，及时记录一下，防止下次安装过程再次入坑，也希望能够帮助到更多的人。
• 说明：对于刚装完的系统是不带Nvidia显卡驱动的，为了方便后面操作，记住先不要用系统的软件升级安装第三方驱动！！！【如果安装了的话，再进行后面操作前清先卸载与英伟达驱动相关的东西，不然后面会失败】

步骤

一、安装Cuda

• 下载地址：https://developer.nvidia.com/cuda-toolkit-archive
• Cuda中带有Nvidia-Driver的，因此无需单独安装驱动，另外Cuda和Driver有版本对应关系，因此，使用Cuda中的Driver能够保证安装的正常
• 采用run文件的安装方法安装Cuda
  • 1、卸载已有的Nvidia 驱动和cuda【如果没有安装过则跳过这一步】

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

  • 2、禁用nouveau驱动

    sudo vi /etc/modprobe.d/blacklist-nouveau.conf
    # 在文件中输入一下内容：
    blacklist nouveau
    options nouveau modeset=0
    # 执行更新
    sudo update-initramfs –u
    # 验证（没有信息输出表示成功，可能需要reboot重启以下电脑）
    lsmod | grep nouveau
    

  • 3、Ctrl+F3进入文本模式【非图形界面（中文会出现乱码，但不影响）】【Ctrl+F2恢复图形界面】
  • 4、关闭图形界面【似乎可以忽略】

    sudo service lightdm stop 
    

  • 5、运行安装程序【选择同时安装Driver和Cuda】

    sudo sh XXX.run
    

  • 6、添加路径

    sudo vim ~/.bashrc
    # 在文件末尾添加一些信息
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
    export PATH=$PATH:/usr/local/cuda-11.1/bin
    export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1
    # 刷新配置
    source  ~/.bashrc
    

  • 7、nvcc -V测试（安装成功，则会输出版本号）
  • 8、启动图形界面【似乎可以忽略】
  • 9、切换回图形界面：Ctrl+F2
    

二、安装Cudnn

• 官方文档：https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
• 下载地址：https://developer.nvidia.com/rdp/cudnn-download
• 需要下载的文件【四个】：第2/4/5/6个文件
  -
• 解压文件：tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
• 复制文件到Cuda目录：

   sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
   sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
  

• 安装deb文件

  sudo dpkg -i libcudnn8_x.x.x-1+cudax.x_amd64.deb
  sudo dpkg -i libcudnn8-dev_8.x.x.x-1+cudax.x_amd64.deb
  sudo dpkg -i libcudnn8-samples_8.x.x.x-1+cudax.x_amd64.deb
  

• 验证：
  • 复制测试样例文件到可写的文件夹
    cp -r /usr/src/cudnn_samples_v8/ $HOME
  • 进入文件夹：
    cd $HOME/cudnn_samples_v8/mnistCUDNN
  • 编译：make clean && make
  • 运行测试样例：./mnistCUDNN【显示test passed说明安装成功！】

由于安装过程没有及时截图，一些细节可以参考另一位博主的结果：https://www.cnblogs.com/booturbo/p/13960935.html