Ubuntu 20.4 安装 nvidia-driver 和 cuda

系统 Ubuntu 20.04

显卡 A5000

1. 安装驱动

1.1 清理已安装nvidia-driver和cuda-清理完后一定要重启

#清理 nvidia-driver
sudo apt-get remove --purge '^nvidia-.*'
sudo apt-get --purge remove "*nvidia*"
sudo apt-get install ubuntu-desktop
sudo rm /etc/X11/xorg.conf
echo 'nouveau' | sudo tee -a /etc/modules
sudo apt-get autoremove
sudo apt-get autoclean
sudo apt-get update

#清理 cuda
sudo apt-get --purge remove "*cublas*" "cuda*" "nsight*" 
sudo rm -rf /usr/local/cuda*
sudo apt-get autoremove
sudo apt-get autoclean
sudo apt-get update

##重启电脑
sudo reboot

1.2 安装驱动，这里我们安装 470版本

sudo apt install nvidia-driver-470 nvidia-dkms-470
#安装完后重启
sudo reboot

2. 安装cuda

cuda我们选择11.4版本，实际上470显卡支持的最大版本也为11.4.

登陆网址CUDA Toolkit 11.4 Downloads | NVIDIA Developer, 照着下边依次选择。 

运行代码，下载 并运行.run文件，开始安装。

#下载
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run

#运行
sudo sh cuda_11.4.0_470.42.01_linux.run

进入如下安装页面：

注意： 这里摁回车取消‘Driver’安装，否则会再次安装470显卡驱动。然后选择‘install’即可 

安装完成以后将cuda添加进系统路径。终端输入代码进入.bashrc后

gedit ~/.basrch

在最下边添加以下两行

export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"

 3. 验证安装结果

#验证驱动安装
nvidia-smi
#验证cuda安装
nvcc -V