CUDA学习笔记(一)Hello World From GPU

编程入门 行业动态 更新时间:2024-10-18 14:19:01

CUDA<a href=https://www.elefans.com/category/jswz/34/1770117.html style=学习笔记(一)Hello World From GPU"/>

CUDA学习笔记(一)Hello World From GPU

本篇博文转载于/,仅用于学习。

准备

如果你是第一次使用CUDA,在Linux下可以使用下面的命令来检查CUDA编译器是否安装正确:

which nvcc

一般,该指令输出为:

/usr/local/cuda/bin/nvcc

另外,你可能还需要检查下你机器上的GPU型号,可以使用给下面的命令查询:

ls  -l /dev/nv*

可能的输出为:

crw-rw-rw- 1 root root 195, 0 Jul 3 13:44 /dev/nvidia0

crw-rw-rw- 1 root root 195, 1 Jul 3 13:44 /dev/nvidia1

crw-rw-rw- 1 root root 195, 255 Jul 3 13:44 /dev/nvidiactl

crw-rw---- 1 root root 10, 144 Jul 3 13:39 /dev/nvram

以上输出显示这里有两个GPU显卡安装在机器上。

写一段CUDA程序的基本过程为:

  • 创建源文件,以”.cu”为后缀。
  • 用nvcc编译程序。
  • 命令行运行。

代码

首先,为了对比,先写一段简单的C程序输出hello world:

#include <stdio.h>iint main(void){printf(“helllo world!\n”);}

然后是我们所谓的kernel function(即CUDA代码):

__global__ void helloFromGpu(void){printf(“hello world form GPU!\n”);}

限定符__global__告诉编译器这个function将由CPU调用在GPU上执行,其调用形式为:

helloFromGPU<<<1,10>>>();

一个kernel是由一组线程执行,所有线程执行相同的代码。上面一行三对尖括号中的1和10 表明了该function将有10个线程,具体含义之后博文中会详述。下面是完整代码:

__global__ void helloFromGPU (void)
{printf(“Hello World from GPU!\n”);
}int main(void)
{
// hello from cpu
printf(“Hello World from CPU!\n”);helloFromGPU <<<1, 10>>>();cudaDeviceReset();return 0;
}

 

这里顺便提及下,我们将CPU端称为host,GPU端称为device。

cudaDeviceReset()用来显式的摧毁清理CUDA程序占用的资源。现在用下面的命令编译:

nvcc –arch sm_20 hello.cu –o hello

 

-arch sm_20是用来指定编译器使用Fermi架构产生device代码。编译成功后执行

./hello

Hello World from CPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

Hello World from GPU!

一个典型的CUDA程序结构包含五个主要步骤:分配GPU空间。将数据从CPU端复制到GPU端。调用CUDA kernel来执行计算。计算完成后将数据从GPU拷贝回CPU。清理GPU内存空间。

更多推荐

CUDA学习笔记(一)Hello World From GPU

本文发布于:2023-12-05 20:47:07,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1665249.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:学习笔记   CUDA   GPU   World

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!