CUDA
的完整形式是计算统一设备架构。 CUDA
是NVIDIA
公司开发的并行计算平台和编程模型。它用于在 NVIDIA
图形处理单元 (GPU) 上运行程序,以显着加快计算应用程序的速度。
在 Ubuntu 上安装 NVIDIA 驱动程序
必须在 Ubuntu 操作系统上安装 NVIDIA GPU 驱动程序才能使 CUDA 正常工作。如果您尚未在 Ubuntu 计算机上安装 NVIDIA GPU 驱动程序并且需要任何帮助来执行此操作,请阅读有关在 Ubuntu 22.04 LTS 上安装 NVIDIA 驱动程序的文章。
nvidia-smi
更新 APT 包存储库缓存
在 Ubuntu 上安装 NVIDIA 驱动程序后,请使用以下命令更新 APT 软件包存储库缓存:
sudo apt update
Ubuntu 的 APT 软件包存储库缓存应该更新。
安装 GCC 和其他构建工具
要编译 CUDA 程序,您需要在 Ubuntu 计算机上安装 GCC 和其他一些构建工具。
要在 Ubuntu 上安装 GCC 编译器和所需的构建工具,请运行以下命令:
sudo apt install build-essential
要确认安装,请按Y,然后按。
现在正在下载 GCC 和所需的软件包。需要一段时间才能完成。
现在正在安装 GCC 和所需的软件包。需要一段时间才能完成。
此时应安装 GCC 和 CUDA 工作所需的构建工具。
要检查是否可以访问 GCC C 和 C++ 编译器,请运行以下命令:
gcc --version
g++ --version
从官方 Ubuntu 软件包存储库在 Ubuntu 上安装 CUDA
要从 Ubuntu 官方软件包存储库安装 CUDA,请运行以下命令:
sudo apt install nvidia-cuda-toolkit nvidia-cuda-toolkit-gcc
要确认安装,请按Y,然后按。
CUDA 和所需的软件包正在下载。需要一段时间才能完成。
CUDA 和所需的软件包正在安装。需要一段时间才能完成。
此时应该安装 CUDA。
测试Ubuntu上CUDA是否安装成功
要检查 CUDA 在 Ubuntu 上是否安装成功,请运行以下命令:
nvcc --version
我们的 Ubuntu 机器上安装了 CUDA 11.5 版本。
编写、编译和运行简单的 CUDA 程序
现在您已经在 Ubuntu 22.04 LTS 计算机上安装了 CUDA,我们将向您展示如何编写、编译和运行一个非常简单的 CUDA“hello world”程序。
首先,创建一个新的“hello.cu
”文件(如果您想继续,则在 ~/codes 目录中)。使用您选择的代码编辑器打开它,然后输入以下代码行:
注意:CUDA 源文件以“.cu”扩展名结尾。
#include <stdio.h>
__global__ void sayHello() {
printf("Hello world from the GPU!\n");
}
int main() {
printf("Hello world from the CPU!\n");
sayHello<<<1,1>>>();
cudaDeviceSynchronize();
return 0;
}
使用“nvcc”CUDA编译器编译“hello.cu”CUDA程序并创建可执行文件“hello”,请运行以下命令:
nvcc hello.cu -o hello
您可以运行编译后的“hello”CUDA程序,如下所示:
./hello
如果看到以下输出,则说明 CUDA 在您的 Ubuntu 计算机上运行良好。编译和运行 CUDA 程序应该没有问题: