Ubuntu 安装 NVIDIA 驱动、CUDA 和 cuDNN：全面指南

引言

NVIDIA 驱动程序、CUDA 和 cuDNN 是深度学习和人工智能应用程序必不可少的组件。通过利用 NVIDIA GPU 的强大功能，这些工具可以大幅提高训练和推理模型的效率。本文旨在为 Ubuntu 用户提供分步指南，帮助他们安装和配置这些必备组件。

先决条件

在继续安装过程之前，请确保满足以下先决条件：

• Ubuntu 18.04 或更高版本（64 位）
• NVIDIA 图形卡（支持 CUDA）
• 互联网连接

第 1 步：下载 NVIDIA 驱动程序

从 NVIDIA 官方网站下载最新的 NVIDIA 驱动程序。选择与您的 GPU 型号和 Ubuntu 版本相对应的驱动程序。

第 2 步：安装 NVIDIA 驱动程序

1. 授予终端 root 权限：sudo su
2. 运行以下命令：sh NVIDIA-Linux-x86_64-*.run
3. 按照提示完成安装。

第 3 步：下载 CUDA

从 NVIDIA 开发者网站下载最新版本的 CUDA。选择与您的 GPU 型号和 Ubuntu 版本相对应的 CUDA 工具包。

第 4 步：安装 CUDA

1. 解压缩下载的 CUDA 工具包：tar -xzvf cuda_*.tar.gz
2. 进入解压缩后的目录：cd cuda-*/
3. 运行以下命令：sudo ./cuda-install.sh
4. 按照提示完成安装。

第 5 步：下载 cuDNN

从 NVIDIA 开发者网站下载最新版本的 cuDNN。选择与您的 CUDA 版本相对应的 cuDNN 库。

第 6 步：安装 cuDNN

1. 解压缩下载的 cuDNN 库：tar -xzvf cudnn-*.tgz
2. 复制 cuDNN 头文件和库：
   • sudo cp cuda/include/cudnn.* /usr/local/cuda/include
   • sudo cp cuda/lib64/libcudnn.* /usr/local/cuda/lib64
3. 创建符号链接：sudo ln -s /usr/local/cuda/include/cudnn.h /usr/include/cudnn.h

第 7 步：验证安装

要验证安装是否成功，请运行以下命令：

nvcc -V

如果命令输出显示 CUDA 和 cuDNN 版本，则表示安装成功。

结论

通过遵循本指南中的步骤，您已成功在 Ubuntu 系统上安装了 NVIDIA 驱动程序、CUDA 和 cuDNN。现在，您的系统已准备好在深度学习和人工智能应用程序中充分利用 NVIDIA GPU 的功能。如果您在任何步骤中遇到问题，请随时参考 NVIDIA 官方文档或在论坛上寻求帮助。