CMake中使用CUDA的方法

简介

Cmake可以管理大型C++项目，为了使用CUDA开展GPU并行计算，需要掌握如何进行配置

CMake代码

将C++代码保存为main.cu，在同一目录下新建CMakeLists.txt文件，输入以下内容

cmake_minimum_required(VERSION 3.0)
project(CUDA_EQUATION)

find_package(CUDA REQUIRED)

message(STATUS "cuda version: " ${CUDA_VERSION_STRING})
message(STATUS ${CUDA_INCLUDE_DIRS})
include_directories(${CUDA_INCLUDE_DIRS})

cuda_add_executable(CUDA_EQUATION main.cu)
target_link_libraries(CUDA_EQUATION ${CUDA_LIBRARIES})

安装CMake的编译方式编译运行即可

mkdir build
cd build/
cmake ..
make
./CUDA_EQUATION