CUDA学习(二):Cmake编译Hello cuda!

文章目录

    • 一、配置CMakeLists.txt
    • 二、hello cuda!

ubuntu下写Cu代码,那首选IDEL当然是CLion了。
Clion编译Cuda需要CMakeLists.txt

一、配置CMakeLists.txt

0、用CLion新建立一个工程 CUDA Executable,或者直接建立一个普通的C++ Executable将主函数main.cpp变成了main.cu。
CUDA学习(二):Cmake编译Hello cuda!_第1张图片
然后配置CMakeList.txt文件
1、FIND_PACKAGE找到已经安装的CUDA
2、使用CUDA_ADD_LIBRARY取代原来的ADD_LIBRARY
3、如果是可执行程序,请使用CUDA_ADD_EXECUTABLE取代ADD_EXECUTABLE。

cmake_minimum_required(VERSION 3.14)
project(Hello)

set(CMAKE_CXX_STANDARD 14)
find_package(CUDA REQUIRED)

cuda_add_executable(Hello main.cu)

二、hello cuda!

#include "cuda_runtime.h"
#include "device_launch_parameters.h"

#include 

__global__ void test(void)  // __global__ 为CUDA的关键字,表示代码在设备端(GPU端)运行, 可以在CPU端被调用
{
     
    printf("Hello CUDA!\n");
}

int main()
{
     
    test <<<1, 1 >>> ();  // 函数调用,  <<< >>>中的第一个参数表示块的个数, 第二个参数表示每个线程块中线程的个数
    // 这里是使用一个线程块,这个线程块中只有一个线程执行这个函数.
    cudaDeviceSynchronize(); // 会阻塞当前程序的执行,直到所有任务都处理完毕(这里的任务其实就是指的是所有的线程都已经执行完了kernel function)。
    // 通俗讲,就是等待设备端的线程执行完成
    // 一个线程块中可以有多个线程,GPU的线程是GPU的最小操作单元

    return 0;
}

CLion运行即可
CUDA学习(二):Cmake编译Hello cuda!_第2张图片

你可能感兴趣的:(编程语言,CUDA入门)