Quiet
  • 主页
  • 归档
  • 分类
  • 标签
  • 链接
  • 关于我

bajiu

  • 主页
  • 归档
  • 分类
  • 标签
  • 链接
  • 关于我
Quiet主题
  • Linux

在 Ubuntu 22.04 LTS 上安装 CUDA

bajiu
环境搭建

2024-01-31 13:57:00

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 程序应该没有问题:

上一篇

ubuntu安装MongoDB数据库

下一篇

node上传文件乱码

©2024 By bajiu.