【记录】Ubuntu24.04安装Nvidia驱动

对于一台裸机, 若使用其跑AI训练程序, 一般需安装 Nvidia驱动 -> CUDA Toolkits -> cuDNN。由于驱动和软件安装方式一直在改变, 所以本文记录Ubuntu 24.04系统安装上述驱动和软件的过程。

1 Nvidia驱动安装

本文直接使用Ubuntu内置命令行安装, 具体步骤如下：

1.1 安装Nvidia PPA

NVIDIA PPA是Ubuntu系统中的第三方软件源，用于安装NVIDIA显卡驱动。通过添加PPA源，用户可以方便地获取最新版本的驱动，并实现自动化安装。

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

1.2 安装Nvidia驱动

使用 ubuntu-drivers 工具自动检测并安装最佳驱动

sudo ubuntu-drivers autoinstall

或特定的驱动版本，可以直接安装

sudo apt install nvidia-driver-xxx

其中 xxx 是驱动版本号，例如 nvidia-driver-570

1.3 测试安装

命令行输入

nvidia-smi