Ubuntu18.04安装驱动、cuda、cudnn（亲测有效）

1、Ubuntu18.04安装显卡驱动

• 禁用SecureBoot（按照主板型号，查询开机进入bios按键，关闭secureboot）

• 禁用nouveau

  sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
  #在文件中插入以下内容，将nouveau加入黑名单，默认不开启
  blacklist nouveau
  options nouveau modeset=0
  #输入以下命令使禁用生效然后重启
  sudo update-initramfs -u
  sudo reboot
  #重启后验证（输入后没反应即为成功）
  lsmod | grep nouveau
  

• 安装驱动

  #1、安装前保证驱动卸载干净
  sudo ./NVIDIA-Linux-x86_64-384.59.run(修改各自版本号) --uninstall
  sudo apt-get remove --purge nvidia*
  #2、ppa源驱动安装，首先查看电脑最适合的版本
  ubuntu-drivers devices
  #加入ppa源
  sudo add-apt-repository ppa:graphics-drivers/ppa
  sudo apt-get update
  sudo apt-get install nvidia-driver-435 #此处数字要对应上面查询到的版本号
  sudo apt-get install mesa-common-dev
  #安装后重启即可，查看是否安装成功 
  nvidia-smi
  

参考：Ubuntu18.04安装Nvidia显卡驱动教程_知遥君的博客-CSDN博客_ubuntu18.04安装nvidia显卡驱动

2、安装cuda

• 首先进入官网下载对应的cuda版本（注：下载cuda前，要搞清楚显卡的型号，深度学习代码要求的环境，如A6000显卡，使用pytorch只能安装cuda11.1版本以上的，否则运行代码时会提示算力不匹配）

• 选择runfile(local)方式下载安装，（速度快）

• #（注：如显卡驱动已经安装，要取消重复安装显卡，也可以自动配置需要安装的路径，是否覆盖cuda软链接等）
  sudo sh cuda_10.1.243_418.87.00_linux.run
  

• 添加环境变量

  #在home目录的.bashrc中导入cuda的环境变量
  export CUDA_HOME=/usr/local/cuda 
  export PATH=$PATH:$CUDA_HOME/bin 
  export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  #退出.bachrc文件，刷新
  source ~/.bashrc
  

• 输入nvcc -V查看是否安装成功

参考：Ubuntu18.04安装Cuda10.1/Cudnn_知遥君的博客-CSDN博客

3、安装cudnn

• 按照官网确定版本号（如A6000在cudnn8是能加快渲染速度）

• 下载，解压

  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 
  sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
  

4、最后

如有任何问题，欢迎大家评论留言