CMake如何编译CUDA(.cu)源文件

2018-02-09 12:37:42来源:http://www.mobibrw.com/2018/10312作者:默默的点滴人点击

分享


现在的项目,如果需要用到计算加速,
Nvidia

CUDA
往往是首选。那么如何在
CMake
中编译写好的
CUDA
源代码,可以参考如下。



首先使用
FIND_PACKAGE
找到已经安装的
CUDA
,此时需要配置的环境变量等,应该已经自动配置完成了


SET(CUDA_VERSION 8.0)
FIND_PACKAGE(CUDA ${CUDA_VERSION} REQUIRED)
STRING (TOLOWER ${CMAKE_SYSTEM_NAME} LOWER_SYSTEM_NAME)
SET(CUDA_TARGET_INCLUDE  ${CUDA_TOOLKIT_ROOT_DIR}-${CUDA_VERSION}/targets/${CMAKE_HOST_SYSTEM_PROCESSOR}-${LOWER_SYSTEM_NAME}/include)


接下来,使用
CUDA_ADD_LIBRARY
取代原来的
ADD_LIBRARY
,如下:


CUDA_ADD_LIBRARY(mylib SHARED
    file1.cpp
    file2.cu
    file3.cpp
    OPTIONS -arch sm_20
)
 
TARGET_LINK_LIBRARIES(mylib ${CUDA_LIBRARIES})


如果是可执行程序,请使用
CUDA_ADD_EXECUTABLE
取代
ADD_EXECUTABLE



参考链接



CMake: how to add cuda to existing project


最新文章

123

最新摄影

微信扫一扫

第七城市微信公众平台