实战:Hello World——CUDA

写在前面的话:本人刚入坑CUDA编程,大家一起相互交流学习

众所周知,学习编程的第一个demo是啥,那就是"Hello World"。没错本文是我第一个CUDA程序,仅以此文开始入门。哈哈哈 希望不要从入门到放弃。

运行前请先检查环境, 运行以下命令查看CUDA编译器驱动版本

nvcc -V

输出如下:

OK,下面我们直接上“Hello World”代码,并保存为hello_gpu.cu文件, 如下:

#include 
__global__ void  hellofromgpu(void )
{
    printf("Hello World from GPU\n");
}

int main(void )
{
    printf("Hello World from cpu\n");
    hellofromgpu<<<1,10>>>();
    cudaDeviceReset();
    return 0;
}

其中:

修饰符__global__是告诉编译器这个函数将会从编译器中调用,然后在GPU上执行

语句hellofromgpu<<<1,10>>>();是指用10个GPU线程来执行hellofromgpu函数,由此我们不难猜出这个程序会输出10遍“Hello World from GPU”。

语句cudaDeviceReset();用来清空和释放当前进程中与当前设备有关的资源。

我们使用下面语句对上面的程序进行编译:

 nvcc -arch sm_75 hello_gpu.cu  -o hello_gpu

其中:

参数-arch是选择编译器架构,我的显卡是2060,所以是sm_75. 这个需要根据显卡的型号来设置(点击这里查看参数)。

运行成功后可以在该文件夹下看到一个hello_gpu的可执行文件,使用语句运行它:

./hello_gpu

 运行结果如下:

实战:Hello World——CUDA_第1张图片

你可能感兴趣的:(CUDA学习笔记,cuda)