如何在 Ubuntu 20.04 LTS 上安装 CUDA
CUDA的完整形式是计算统一设备架构。 CUDA是NVIDIA公司开发的并行计算平台和编程模型。它用于在 NVIDIA 图形处理单元 (GPU) 上运行程序，以显着加快计算应用程序的速度。
在本文中，我将向您展示如何从 Ubuntu 官方软件包存储库在 Ubuntu 20.04 LTS 上安装 CUDA。我还将向您展示如何从官方 NVIDIA 软件包存储库在 Ubuntu 20.04 LTS 上安装最新版本的 CUDA。我还将向您展示如何编写、编译和运行您的第一个 CUDA 程序。那么，让我们开始吧。
目录
先决条件
更新包存储库缓存
安装 GCC 和其他构建工具
从官方 Ubuntu 软件包存储库安装 CUDA
从官方 NVIDIA 软件包存储库安装最新版本的 CUDA
使用 CUDA 编写 Hello World 程序
结论
参考
先决条件
在开始之前，
i) 您的计算机上必须安装有 NVIDIA GPU。
ii) 您的计算机上必须安装 NVIDIA GPU 驱动程序。
要确认 NVIDIA GPU 驱动程序是否正常工作，请运行以下命令：
如果 NVIDIA 驱动程序内核模块正在工作，您应该看到以下输出。
如果 NVIDIA 驱动程序正常工作，NVIDIA 命令行工具应该按预期工作。
NVIDIA X Server Settings 应用程序等 NVIDIA 图形工具也应该可以使用。
更新包存储库缓存：
满足所有要求后，请使用以下命令更新 APT 软件包存储库缓存：
应更新 APT 包存储库缓存。
安装 GCC 和其他构建工具：
为了使 CUDA 工作，您必须在计算机上安装 GCC 和其他构建工具。
您可以使用以下命令从 Ubuntu 官方软件包存储库安装 GCC 和所有必需的构建工具：
要确认安装，请按Y，然后按
正在下载所有必需的包。需要一段时间才能完成。
下载完软件包后，APT 会一一安装它们。需要一段时间才能完成。
此时，GCC 和所有必需的构建工具应该已安装。
正如您所看到的，GCC 编译器工作得很好。
$ g++ --version
从官方 Ubuntu 软件包存储库安装 CUDA
CUDA 版本 10 可在 Ubuntu 20.04 LTS 的官方软件包存储库中获取。
要从 Ubuntu 20.04 LTS 的官方软件包存储库安装 CUDA v10，请运行以下命令：
要确认安装，请按Y，然后按
所有必需的软件包将从互联网下载。需要一段时间才能完成。
下载所需的软件包后，将一一安装。需要一段时间才能完成。
此时，CUDA 和所有必需的依赖项应已安装。
要确认 CUDA 是否正常工作，请运行以下命令：
从官方 NVIDIA 软件包存储库安装最新版本的 CUDA
在撰写本文时，CUDA 11 是 CUDA 的最新版本。您可以从 NVIDIA 官方软件包存储库安装最新版本的 CUDA。
首先，运行以下命令以确保安装了所需的 Linux 标头。
应安装 Linux 内核头文件。就我而言，它们已经安装了。
现在，使用以下命令从 NVIDIA 官方网站下载 CUDA 存储库 Pin 文件：
要添加官方 NVIDIA 软件包存储库的 GPG 密钥，请运行以下命令：
应将官方 NVIDIA 软件包存储库的 GPG 密钥添加到 APT 软件包管理器中。
要添加官方 NVIDIA CUDA 软件包存储库，请运行以下命令：
应添加官方 NVIDIA CUDA 软件包存储库。
现在，使用以下命令更新 APT 包存储库缓存：
在从官方 NVIDIA 软件包存储库安装最新版本的 CUDA 之前，最好更新 Ubuntu 20.04 LTS 计算机的所有现有软件包。
要更新 Ubuntu 20.04 LTS 计算机的所有现有软件包，请运行以下命令：
要确认更新，请按Y，然后按
所有必需的更新都是从互联网下载的。需要一段时间才能完成。
下载更新后，APT 包管理器将一一安装它们。需要一段时间才能完成。
此时，应安装所有更新。
现在，您应该能够从 NVIDIA 官方软件包存储库安装最新版本的 CUDA，如下所示：
要确认安装，请按Y，然后按
应下载并安装最新版本的 CUDA 以及所有必需的依赖包。需要一段时间才能完成。
此时，最新版本的 CUDA 和所有必需的依赖包应该已安装。
要将 CUDA 添加到路径，请在 /etc/profile.d/ 目录中创建一个新脚本 cuda.sh，如下所示：
在 cuda.sh 脚本中输入以下行。
export PATH="$PATH:$CUDA_HOME/bin"
完成后，按
为了使更改生效，请按如下方式重新启动计算机：
您应该能够访问 CUDA 工具，如下面的屏幕截图所示。
为了运行某些 CUDA 工具，您可能需要超级用户权限。要允许 sudo 以超级用户权限运行 /usr/local/cuda/bin 目录（其中从官方 NVIDIA 软件包存储库安装最新版本的 CUDA）中的二进制文件，您必须配置/etc/sudoers 文件。
使用以下命令打开 /etc/sudoers 配置文件进行编辑：
在 /etc/sudoers 文件中找到标记行。
在 secure_path 的末尾，附加 :/usr/local/cuda/bin，如下面的屏幕截图所示。
完成后，按
从现在开始，如果需要，您应该能够使用 sudo 以超级用户权限运行 CUDA 工具。
使用 CUDA 编写 Hello World 程序
在本节中，我将向您展示如何编写第一个 CUDA 程序。该程序将简单地打印一条来自 GPU 的消息和一条来自 CPU 的消息。如果该程序成功运行，您可以确认 CUDA 正在您的计算机上运行。
创建一个新文件 hello.cu 并输入以下代码行：
注意： CUDA 源文件以扩展名 .cu 结尾
__global__ void say_hello() {
printf("Hello world from the GPU!\n");
}
int main() {
printf("Hello world from the CPU!\n");
say_hello<<>>();
cudaDeviceSynchronize();
return 0;
}
完成后，将 hello.cu 文件保存在 ~/codes 目录或您选择的任何其他目录中。
导航到 ~/codes 目录或保存 hello.cu 文件的目录。
要使用 CUDA 编译器 nvcc 编译 hello.cu CUDA 源文件，请运行以下命令：
CUDA 源文件 hello.cu 应该被编译而没有任何错误，并且应该创建一个新的可执行/二进制文件 hello，如下面的屏幕截图所示。
您可以运行编译好的CUDA程序hello，如下所示：
如果您看到以下输出，则 CUDA 在您的计算机上运行良好。编译和运行 CUDA 程序应该没有问题。
结论
在本文中，我向您展示了如何从 Ubuntu 官方软件包存储库在 Ubuntu 20.04 LTS 上安装 CUDA。我还向您展示了如何从官方 NVIDIA 软件包存储库在 Ubuntu 20.04 LTS 上安装最新版本的 CUDA。我已经向您展示了如何编写第一个 CUDA 程序、编译并运行它。
参考
[1] CUDA 专区 |英伟达开发商
[2] 安装指南 Linux:: CUDA 工具包文档 (nvidia.com)
[3]教程01：向CUDA问好——CUDA教程（cuda-tutorial.readthedocs.io）
[4] 您的第一个 CUDA C 程序 – YouTube
[5] cuda 教程 => 让我们启动一个 CUDA 线程来打个招呼 (riptutorial.com)