Ubuntu下完全移除cuda和nvidia驱动并重新安装新版本cuda

Ubuntu下完全移除cuda和nvidia驱动并重新安装新版本cuda

本文演示的成功安装的设备信息为：

Device	Parameter
GPU	RTX3060
OS	Ubuntu 20.04
NVIDIA Driver	535
GCC	9.4
CUDA Toolkit	12.1
Cudnn	8.7.6.29

1. 完全移除旧版本cuda和nvidia驱动

打开terminal，输入：

dpkg -l | grep -i nvidia

用于查看现有驱动。如果想仅移除nvidia-cuda-toolkit驱动，则输入：

sudo apt-get remove nvidia-cuda-toolkit

如果想移除nvidia-cuda-toolkit及其依赖项，则输入：

sudo apt-get remove --auto-remove nvidia-cuda-toolkit

如果想移除配置数据，则输入：

sudo apt-get purge --auto-remove nvidia-cuda-toolkit

移除所有关于cuda的内容，则输入：

sudo apt-get --purge -y remove 'cuda*'
# cuda10.1及以上的卸载
cd /usr/local/cuda-xx.x/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-xx.x

移除所有关于nvidia的内容，则输入：

sudo apt-get --purge -y remove 'nvidia*'

检查cuda的文件夹：

ls /usr/local/ | grep cuda

随后重启系统：

sudo reboot

2. 移除旧版的cudnn

在terminal中输入：

dpkg -l | grep cudnn

找到系统中存在的cudnn文件，然后输入：

sudo dpkg --remove libcudnn8 libcudnn8-dev libcudnn9-samples

即可移除全部cudnn的旧版本文件（以cudnn8为例）。

3. 重新安装nvidia驱动

首先，upgrade ubuntu：

sudo apt update
sudo apt upgrade 

然后，查看可安装的nvidia驱动列表：

sudo apt install ubuntu-drivers-common
sudo ubuntu-drivers devices

选取一个列表中建议的nvidia驱动按本，如nvidia-driver-535进行安装：

sudo ubuntu-drivers autoinstall

随后重启系统：

sudo reboot

在terminal输入：

nvidia-smi

查看nvidia驱动的安装情况以及cuda最高可安装的版本。目前本机安装的情况为：

NVIDIA-SMI 535.129.03   Driver Version: 535.129.03   CUDA Version: 12.2

安装gcc并查看gcc版本：

sudo apt install gcc
gcc -v

4. 安装cuda toolkit

打开cuda toolkit官网，查找所需要的版本，可以采用线上安装的方式 (以cuda12.1为例)。进入下载文件夹，打开terminal输入：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu2004-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

然后重启系统。之后输入：

sudo gedit ~/.bashrc

将以下内容复制进文档最后：

export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存之后更新文件：

source ~/.bashrc

5. 安装cudnn

打开cudnn官网（需要注册），下载所需要的cudnn.deb文件，然后输入：

sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_amd64.deb

其中$distro是系统版本，如ubuntu2004。然后导入CUDA GPG key：

sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/

然后更新数据库元数据：

sudo apt-get update

最后安装runtime library, developer library和code samples：

sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y

其中8.x.x.x为cudnn版本号，cudaX.Y为cuda版本号。如果此命令失败，则运行以下命令：

sudo apt-get install libcudnn8 libcudnn8-dev libcudnn8-samples

输入nvcc -V即可查看安装cuda的版本，正常安装后应该显示：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Feb__7_19:32:13_PST_2023
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0