Ubuntu安装深度学习模型驱动的快捷步骤

在 Ubuntu 服务器上安装 NVIDIA RTX 显卡的驱动程序以进行深度学习涉及一系列的步骤，这些步骤包括准备操作系统、安装合适的 NVIDIA 驱动程序，以及安装 CUDA 和 cuDNN 库。以下是详细的步骤：

步骤 1: 准备系统

1. 更新你的系统：
   打开终端并运行以下命令来更新你的 Ubuntu 系统：

   sudo apt update
   sudo apt upgrade
   

2. 安装内核头文件和构建工具：
   这对于后续安装驱动是必要的。

   sudo apt install build-essential linux-headers-$(uname -r)
   

步骤 2: 禁用 Nouveau 驱动

Nouveau 是 NVIDIA 显卡的开源驱动，它与 NVIDIA 官方驱动通常不兼容。禁用它是推荐的步骤：

1. 编辑黑名单配置文件：

   sudo nano /etc/modprobe.d/blacklist-nouveau.conf
   

2. 添加以下行到文件中：

   blacklist nouveau
   options nouveau modeset=0
   

3. 更新初始化内核并重启：

   sudo update-initramfs -u
   sudo reboot
   

步骤 3: 安装 NVIDIA 驱动

1. 添加 NVIDIA PPA（非官方PPA，但提供了最新的驱动）：

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

2. 安装驱动：
   查找最新的驱动版本，4090 通常需要最新或接近最新版本的驱动。你可以通过以下命令查找并安装合适的驱动：

   ubuntu-drivers devices
   sudo ubuntu-drivers autoinstall
   

   或者直接安装推荐的驱动：

   sudo apt install nvidia-driver-最新版本
   

步骤 4: 安装 CUDA 和 cuDNN

1. 下载并安装 CUDA：
   访问 NVIDIA CUDA Toolkit 网站，选择适合 Ubuntu 的版本，并按照页面上的指南下载并安装 CUDA。

2. 下载并安装 cuDNN：
   访问 NVIDIA cuDNN 页面（需要 NVIDIA 开发者账号），下载适合你的 CUDA 版本的 cuDNN，并按照官方文档安装。

步骤 5: 验证安装

1. 重启计算机：

   sudo reboot
   

2. 检查 NVIDIA 驱动安装状态：

   nvidia-smi
   

   这个命令会显示你的 GPU 状态和驱动版本，确认是否正确安装。

3. 测试 CUDA：
   通过运行 CUDA 自带的示例来测试 CUDA 是否正确安装。