一文解决ubuntu20上3060显卡驱动,cuda,cudnn,torch的安装和卸载

---

前言

本文主要介绍在ubuntu20下根据自己的python版本来确定torch版本，从而确定cuda和cudnn版本完成安装（注意，本文可能会同时看到cuda11和cuda10，因为是第一次安装cuda10在3060显卡跑不起来就换成cuda11了），本文版本为python-3.8，torch-1.8.0，torchvision-0.9.0，cuda-11.1.1，cudnn-8.9.0。

---

1、显卡驱动安装

（1）通过以下命令查看匹配的驱动：

ubuntu-drivers devices

从上图看到recommended 470，故安装470版本的驱动

（2）驱动安装方法1，

sudo ubuntu-drivers autoinstall 

（3） 推荐 驱动安装方法2，进入官网下载对应的版本

（4）安装相关依赖,

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

（5） 禁用系统默认显卡驱动，打开文件

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

在文件末尾输入，

blacklist nouveau
options nouveau modeset=0

保存(ctrl+s)后手动更新；

sudo update-initramfs -u

重启后输入以下命令，无输出则禁用成功，

lsmod | grep nouveau

（6）配置环境变量，gedit命令打开配置文件：

sudo gedit ~/.bashrc

在文件末尾输入，

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH

然后，通过下面命令更新以下，

source ~/.bashrc

（7）安装NVIDIA驱动

sudo chmod  a+x NVIDIA-Linux-x86_64-530.41.03