如何在MITK插件中使用CUDA？

Question

我正在尝试在MITK平台上使用CUDA。我使用的MITK平台是2014年版本，它不支持数据自动化系统；然而，我发现了一个回复，它显示了一个解决方案，但它也不适合我。根据MITK的Cmake，如果我在*.cpp脚本中将“CPP_FILES”文件设置为CPP_FILES标志，gcc会编译源代码，但是对于files.cmake没有变量。因此，根据上面的链接，我修改了"mitkFunctionCreateModule.cmake“

if (IS_CUDA)
    find_package(CUDA REQUIRED)
    set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}; -std=c++11 )
    #list(APPEND CUDA_NVCC_FLAGS "-arch=sm_20;-std=c++11;-O2;-DVERBOSE")
    #SET(CUDA_PROPAGATE_HOST_FLAGS OFF)
    message ("is_cuda == true")
    cuda_add_library(${MODULE_TARGET} STATIC ${CUDA_FILES}) 
endif()

在files.cmake中，我添加了以下几行：

find_package(CUDA)
set(CUDA_FILES
  src/robot/cudafolder/testcuda.cu
  src/robot/cudafolder/testcuda.h
)
set(IS_CUDA true)
MITK_CREATE_MODULE(my_module
    DEPENDS MitkCore
    PACKAGE_DEPENDS Qt5|Core
    #ADDITIONAL_LIBS abc glew
)

libmy_module.a是在bin文件夹中创建的，但是我不能将它链接到主可执行文件！

PS:我面临的最大问题是我不能使用target_link_libraries！由于任何我不知道的原因，不可能将这个库链接到主要的MITK文件。任何在插件模块工作的专家都能在这件事上帮我吗？

谢谢

Answer 1

最后，我找到了一个在MITK插件核心添加CUDA的解决方案。

解决方案是在files.cmake的末尾添加这些行

find_package(CUDA)
set(CUDA_FILES
  src/robot/cudafolder/testcuda.cu
  src/robot/cudafolder/testcuda.h)

set(MyCudalib CudaTestLib)
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}; -std=c++11 )
cuda_add_library( ${MyCudalib} SHARED ${CUDA_FILES}) 

然后从边CmakeLists.txt中的相同路径中添加以下行

TARGET_LINK_LIBRARIES(${PROJECT_NAME}  ${MyCudalib} ${CUDA_LIBRARIES} ${CUDA_nppi_LIBRARY})

对我来说很管用