cmake: add support for Nvidia GPU Tesla K80

9 years ago · 3335963987
parent 671e95d59e
commit 3335963987
1 changed files with 3 additions and 3 deletions
--- a/cmake/OpenCVDetectCUDA.cmake
+++ b/cmake/OpenCVDetectCUDA.cmake
@ -65,7 +65,7 @@ if(CUDA_FOUND)
  if(CUDA_GENERATION STREQUAL "Fermi")
    set(__cuda_arch_bin "2.0")
  elseif(CUDA_GENERATION STREQUAL "Kepler")
-    set(__cuda_arch_bin "3.0 3.5")
+    set(__cuda_arch_bin "3.0 3.5 3.7")
  elseif(CUDA_GENERATION STREQUAL "Maxwell")
    set(__cuda_arch_bin "5.0")
  elseif(CUDA_GENERATION STREQUAL "Pascal")
@ -92,9 +92,9 @@ if(CUDA_FOUND)
      set(__cuda_arch_ptx "")
    else()
      if(${CUDA_VERSION} VERSION_LESS "8.0")
-        set(__cuda_arch_bin "2.0 3.0 3.5 5.0")
+        set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0")
      else()
-        set(__cuda_arch_bin "2.0 3.0 3.5 5.0 6.0")
+        set(__cuda_arch_bin "2.0 3.0 3.5 3.7 5.0 6.0")
      endif()
      set(__cuda_arch_ptx "")
    endif()