From a66fd527b05d3e6cfc6ea68a4156dab8b6ea18c2 Mon Sep 17 00:00:00 2001 From: jasjuang Date: Mon, 15 Oct 2018 10:40:24 -0700 Subject: [PATCH] add support for latest Turing gpu and cuda 10 --- cmake/OpenCVDetectCUDA.cmake | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/cmake/OpenCVDetectCUDA.cmake b/cmake/OpenCVDetectCUDA.cmake index 76e57c8fc9..ee4e7cea64 100644 --- a/cmake/OpenCVDetectCUDA.cmake +++ b/cmake/OpenCVDetectCUDA.cmake @@ -81,6 +81,8 @@ if(CUDA_FOUND) set(__cuda_arch_bin "6.0 6.1") elseif(CUDA_GENERATION STREQUAL "Volta") set(__cuda_arch_bin "7.0") + elseif(CUDA_GENERATION STREQUAL "Turing") + set(__cuda_arch_bin "7.5") elseif(CUDA_GENERATION STREQUAL "Auto") execute_process( COMMAND "${CUDA_NVCC_EXECUTABLE}" ${CUDA_NVCC_FLAGS} "${OpenCV_SOURCE_DIR}/cmake/checks/OpenCVDetectCudaArch.cu" "--run" WORKING_DIRECTORY "${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeTmp/" @@ -105,7 +107,7 @@ if(CUDA_FOUND) ERROR_QUIET OUTPUT_STRIP_TRAILING_WHITESPACE) if(NOT _nvcc_res EQUAL 0) message(STATUS "Automatic detection of CUDA generation failed. Going to build for all known architectures.") - set(__cuda_arch_bin "5.3 6.2 7.0") + set(__cuda_arch_bin "5.3 6.2 7.0 7.5") else() set(__cuda_arch_bin "${_nvcc_out}") string(REPLACE "2.1" "2.1(2.0)" __cuda_arch_bin "${__cuda_arch_bin}") @@ -114,8 +116,10 @@ if(CUDA_FOUND) else() if(${CUDA_VERSION} VERSION_LESS "9.0") set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 5.2 6.0 6.1") - else() + elseif(${CUDA_VERSION} VERSION_LESS "10.0") set(__cuda_arch_bin "3.0 3.5 3.7 5.0 5.2 6.0 6.1 7.0") + else() + set(__cuda_arch_bin "3.0 3.5 3.7 5.0 5.2 6.0 6.1 7.0 7.5") endif() endif() endif()