Ubuntu系统中如何将CUDA从11.5升级到12.2

要将 CUDA 从 11.5 升级到 12.2，需要先彻底卸载旧版本，再安装匹配版本的 CUDA 工具包，步骤如下：

1. 卸载现有 CUDA 11.5 及相关组件

# 卸载 CUDA 工具包（11.5 版本）
sudo apt --purge remove "cuda*11.5*" "nvidia*11.5*"

# 卸载残留的 CUDA 相关包
sudo apt --purge remove nvidia-cuda-toolkit
sudo apt autoremove
sudo apt autoclean

# 删除旧版本 CUDA 目录（若存在）
sudo rm -rf /usr/local/cuda-11.5
sudo rm -rf /usr/local/cuda  # 这是旧版本的软链接，需删除

2. 安装 CUDA 12.2（与当前驱动匹配）

你的驱动版本（535.247.01）支持 CUDA 12.2，直接安装对应版本：

步骤 1：添加 NVIDIA 官方 CUDA 12.2 仓库

# 下载并添加 CUDA 12.2 仓库密钥
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600

# 下载 CUDA 12.2 本地安装包
wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda-repo-ubuntu2204-12-2-local_12.2.0-535.54.03-1_amd64.deb

# 安装仓库包并导入密钥
sudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.0-535.54.03-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/

# 更新 apt 缓存
sudo apt update

步骤 2：安装 CUDA 12.2 工具包

# 安装 CUDA 12.2 核心组件（包括 nvcc 编译器）
sudo apt -y install cuda-12-2

3. 配置环境变量（指向新安装的 CUDA 12.2）

# 编辑环境变量配置文件
nano ~/.bashrc

# 在文件末尾添加以下内容（替换旧的 CUDA 路径）
export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# 保存并退出（按 Ctrl+O 保存，Ctrl+X 退出）

# 使配置生效
source ~/.bashrc

4. 验证升级结果

# 检查 nvcc 版本（应显示 12.2）
nvcc --version

# 检查 CUDA 安装路径（应指向 12.2）
ls -l /usr/local | grep cuda  # 应显示 cuda -> cuda-12.2

若输出类似以下内容，说明升级成功：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Mar_21_16:59:10_PDT_2023
Cuda compilation tools, release 12.2, V12.2.91
Build cuda_12.2.r12.2/compiler.32965470_0

补充说明

• 若升级后仍显示旧版本，可能是环境变量未生效，可重启终端或系统解决。
• CUDA 12.2 与你的 RTX A6000 显卡完全兼容，支持新的 CUDA 特性和性能优化。
• 后续若需安装 cuDNN 等库，需选择与 CUDA 12.2 匹配的版本（如 cuDNN 8.9+）。