如何在 Ubuntu 20.04 LTS 上安装 CUDA

CUDA 代表计算统一设备架构。CUDA 包括 CUDA 工具包（编译器、配置文件和调试器）、软件驱动程序和 CUDA SDK。软件层可以直接访问 GPU 的虚拟指令集和并行计算元素。对于深度学习研究和框架开发人员，使用 cuDNN 进行高性能 GPU 加速。

在 Ubuntu 20.04 LTS Focal Fossa 上安装 CUDA

步骤 1. 首先，通过apt在终端中运行以下命令确保所有系统包都是最新的。

sudo apt update
sudo apt upgrade

步骤 2. 在 Ubuntu 20.04 上安装 CUDA。

默认情况下，CUDA 包在 Ubuntu 20.04 基础存储库上不可用。现在我们使用以下命令将 CUDA 存储库 pin 添加到您的系统：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

接下来，我们将引脚移动到目录并将其重命名为：/etc/preferences.dcuda-repository-pin-600

sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

之后，使用以下命令下载 CUDA Ubuntu 文件：

wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

然后，使用dpkg命令执行文件：

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

最后，软件包列出并安装CUDA软件包：

sudo apt update
sudo apt install cuda

步骤 3. 安装 CuDNN。

现在我们从官方页面下载安装的CuDNN。注册 NVIDIA Developer Program 后，您可以下载最新版本的 cuDNN：

tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

接下来，将以下文件复制到 CUDA 目录：

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

感谢您使用本教程在 Ubuntu 20.04 LTS Focal Fossa 系统上安装 CUDA NVIDIA。如需其他帮助或有用信息，我们建议您查看NVIDIA 官方网站。