返回

Debian 12 上安装 NVIDIA GPU Driver 以构建高效的机器学习环境

后端

1. 系统准备

在开始安装 NVIDIA GPU 驱动程序之前，请确保您的 Debian 12 系统已满足以下先决条件：

  • 确认您的系统已安装 Debian 12 操作系统。
  • 确保您的系统已连接到互联网，以便下载必要的软件包。
  • 您拥有系统管理员权限，以便执行安装操作。

2. 添加 NVIDIA 存储库

为了能够安装 NVIDIA GPU 驱动程序，您需要将 NVIDIA 存储库添加到您的系统中。请按照以下步骤进行操作：

  1. 打开终端并输入以下命令：
sudo nano /etc/apt/sources.list
  1. 在打开的文本编辑器中，添加以下行：
deb https://packages.nvidia.com/10 /
  1. 保存并关闭文本编辑器。

3. 导入 NVIDIA 密钥

接下来，您需要导入 NVIDIA 密钥以验证软件包的完整性。请按照以下步骤进行操作：

  1. 打开终端并输入以下命令：
sudo wget https://packages.nvidia.com/10 /pubkey.gpg
  1. 将下载的密钥导入系统：
sudo apt-key add pubkey.gpg
  1. 确认密钥已成功导入：
sudo apt-key list

4. 更新系统软件包

在继续安装之前，请更新系统软件包以确保您拥有最新的软件版本。请按照以下步骤进行操作：

sudo apt update

5. 安装 NVIDIA GPU 驱动程序

现在，您可以安装 NVIDIA GPU 驱动程序了。请按照以下步骤进行操作：

sudo apt install nvidia-driver
  1. 重启系统**

在完成安装后，您需要重启系统以使更改生效。请按照以下步骤进行操作：

sudo reboot

7. 验证安装

为了验证 NVIDIA GPU 驱动程序是否已成功安装，请按照以下步骤进行操作：

  1. 打开终端并输入以下命令：
nvidia-smi
  1. 如果命令输出显示了您的 NVIDIA GPU 信息，则说明驱动程序已成功安装。

8. 安装 CUDA 和 cuDNN

CUDA 和 cuDNN 是用于机器学习的两个重要库。如果您打算使用 GPU 进行机器学习，则需要安装这两个库。请按照以下步骤进行操作：

  1. 安装 CUDA：
sudo apt install cuda
  1. 安装 cuDNN：
sudo apt install libcudnn8

9. 配置环境变量

为了让机器学习框架能够找到 CUDA 和 cuDNN，您需要配置环境变量。请按照以下步骤进行操作：

  1. 打开终端并输入以下命令：
sudo nano ~/.bashrc
  1. 在打开的文本编辑器中，添加以下行：
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

  1. 保存并关闭文本编辑器。

  2. 使更改生效：

source ~/.bashrc

10. 测试安装

现在，您可以测试您的安装是否成功。请按照以下步骤进行操作：

  1. 打开终端并输入以下命令：
python -c "import torch; print(torch.cuda.is_available())"
  1. 如果输出结果为 True，则说明您的安装已成功。

结论

恭喜您，您已成功在 Debian 12 系统上安装了 NVIDIA GPU 驱动程序以及必要的库。现在，您可以开始构建高效的机器学习环境，利用 GPU 加速来提升您的机器学习项目效率。如果您在安装过程中遇到任何问题，请随时寻求帮助。祝您在机器学习领域取得丰硕成果！