diff --git a/itk-module-init.cmake b/itk-module-init.cmake index f8a52d5..537cf78 100644 --- a/itk-module-init.cmake +++ b/itk-module-init.cmake @@ -29,14 +29,17 @@ if(CMAKE_CUDA_COMPILER) endif() # Configure CUDA compilation options -option( - CUDACOMMON_CUDA_VERSION - "Specify the exact CUDA version that must be used for CudaCommon" -) +if( NOT CUDACOMMON_CUDA_VERSION) + set(CUDACOMMON_CUDA_VERSION ${CUDAToolkit_VERSION} CACHE STRING "Specify the exact CUDA version that must be used for CudaCommon") +else() + set(CUDACOMMON_CUDA_VERSION ${CUDACOMMON_CUDA_VERSION} CACHE STRING "Specify the exact CUDA version that must be used for CudaCommon") +endif() +mark_as_advanced(CUDACOMMON_CUDA_VERSION) + if(CUDAToolkit_FOUND) enable_language(CUDA) set(CMAKE_CUDA_RUNTIME_LIBRARY Static) if(CUDACOMMON_CUDA_VERSION) - find_package(CUDAToolkit EXACT ${CUDACOMMON_CUDA_VERSION}) + find_package(CUDAToolkit EXACT ${CUDACOMMON_CUDA_VERSION} REQUIRED) endif() endif()