Ubuntu 系统上安装 NVIDIA CUDA 的详细步骤

以下是在 Ubuntu 系统上安装 NVIDIA CUDA 的详细步骤，分为 驱动安装 和 CUDA Toolkit 安装 两个部分：

一、准备工作

1. 确认GPU型号支持CUDA
   运行以下命令查看NVIDIA GPU型号：GeForce RTX 2080 Ti

   root@i-28e6iose:/home/ubuntu# lspci | grep -i nvidia
   00:0c.0 VGA compatible controller: NVIDIA Corporation TU102 [GeForce RTX 2080 Ti] (rev a1)
   00:0d.0 Audio device: NVIDIA Corporation TU102 High Definition Audio Controller (rev a1)
   

确保你的GPU在NVIDIA CUDA支持列表中。

2. 检查系统版本
   Ubuntu 22.04/20.04 等主流版本均支持，建议使用 LTS（长期支持版）。

3. 卸载旧驱动（如有）
   如果之前安装过NVIDIA驱动或CUDA，先彻底卸载：

   sudo apt purge nvidia-* cuda-* 
   sudo reboot
   

二、安装NVIDIA驱动

方法1：通过APT仓库自动安装（推荐）

1. 添加官方驱动仓库并更新：

   sudo add-apt-repository ppa:graphics-drivers/ppa
   sudo apt update
   

2. 查找推荐的驱动版本：

   ubuntu-drivers devices
   

3. 安装推荐的驱动（例如 nvidia-driver-535）：

   sudo apt install nvidia-driver-535
   

4. 重启系统并验证：

   sudo reboot
   nvidia-smi  # 看到GPU信息表示驱动安装成功
   
   root@i-28e6iose:/home/ubuntu# nvidia-smi
   Fri Feb 28 13:39:55 2025
   +---------------------------------------------------------------------------------------+
   | NVIDIA-SMI 535.183.01             Driver Version: 535.183.01   CUDA Version: 12.2     |
   |-----------------------------------------+----------------------+----------------------+
   | GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
   | Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
   |                                         |                      |               MIG M. |