Question

我正在尝试在MITK平台中使用CUDA。我使用的MITK平台是2014版，不支持CUDA;但是，我找到了一个显示解决方案的reply，但它也不适用于我。根据MITK的Cmake结构，如果我在files.cmake脚本中将“* .cpp”文件设置为CPP_FILES标志，gcc会编译源代码，但是对于cuda，没有变量。因此，根据上面的链接，我修改了“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 exe文件。在插件模块中工作的mitk专家可以帮助我解决这个问题吗？

谢谢

Answer 1

最后，我找到了一个在CITK插件的核心添加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})

它对我有用

如何在MITK插件中使用CUDA？

1 个答案: