diff --git a/DarknetConfig.cmake.in b/DarknetConfig.cmake.in index cbd7f066..c8f4787d 100644 --- a/DarknetConfig.cmake.in +++ b/DarknetConfig.cmake.in @@ -10,7 +10,11 @@ if(@OpenCV_FOUND@) endif() if(@ENABLE_CUDA@) - enable_language(CUDA) + include(CheckLanguage) + check_language(CUDA) + if(NOT CMAKE_CUDA_COMPILER) + message(STATUS " --> WARNING: Unable to find native CUDA integration!") + endif() find_dependency(CUDA) cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS ${CUDA_ARCHITECTURES}) if(@CUDNN_FOUND@)