以下是在 Ubuntu 系统上安装 NVIDIA CUDA 的详细步骤,分为 驱动安装 和 CUDA Toolkit 安装 两个部分:
一、准备工作
-
确认GPU型号支持CUDA
运行以下命令查看NVIDIA GPU型号:GeForce RTX 2080 Tiroot@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支持列表中。
-
检查系统版本
Ubuntu 22.04/20.04 等主流版本均支持,建议使用 LTS(长期支持版)。 -
卸载旧驱动(如有)
如果之前安装过NVIDIA驱动或CUDA,先彻底卸载:sudo apt purge nvidia-* cuda-* sudo reboot
二、安装NVIDIA驱动
方法1:通过APT仓库自动安装(推荐)
-
添加官方驱动仓库并更新:
sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update
-
查找推荐的驱动版本:
ubuntu-drivers devices
-
安装推荐的驱动(例如
nvidia-driver-535
):sudo apt install nvidia-driver-535
-
重启系统并验证:
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. |