cmake+vscode+cuda的使用

1. 环境准备

  • visual studio code
    • 及插件cmake
    • 及插件cmake tools(编译用)
  • visual studio 2017 community编译环境
  • cuda 11.0

注意:1)cmake需要加入环境变量,能通过cmake --version查找到;2)cuda加入环境变量,通过nvcc -V查找到

2. 代码编写

|-CMakeLists.txt

|-main.cu

CMakeLists.txt

cmake_minimum_required(VERSION 3.18)

project(MyCudaProject LANGUAGES CXX CUDA)

find_package(CUDAToolkit REQUIRED)

# 明确指定源文件
set(SOURCE_FILES main.cu)

# 添加可执行文件
add_executable(${PROJECT_NAME} ${SOURCE_FILES})

# 链接 CUDA 库
target_link_libraries(${PROJECT_NAME} PRIVATE CUDA::cudart)

main.cu

#include <stdio.h>
#include <cuda_runtime.h>
__global__ void hello() {
    printf("Hello from GPU!\n");
}

int main() {
    hello<<<1, 1>>>();
    cudaDeviceSynchronize();
    return 0;
}

3. 代码编译

  • 启动编译
    在这里插入图片描述

  • 启动运行
    在这里插入图片描述

Logo

欢迎加入DeepSeek 技术社区。在这里,你可以找到志同道合的朋友,共同探索AI技术的奥秘。

更多推荐