Browse Source

Merge branch 'cuda-arch-version' into release-3.12

Merge-request: !2251
Brad King 7 years ago
parent
commit
c7769867d6
1 changed files with 3 additions and 2 deletions
  1. 3 2
      Modules/FindCUDA/select_compute_arch.cmake

+ 3 - 2
Modules/FindCUDA/select_compute_arch.cmake

@@ -18,8 +18,9 @@
 #
 
 if(CMAKE_CUDA_COMPILER_LOADED) # CUDA as a language
-  if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA")
-    set(CUDA_VERSION "${CMAKE_CUDA_COMPILER_VERSION}")
+  if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA"
+      AND CMAKE_CUDA_COMPILER_VERSION MATCHES "^([0-9]+\\.[0-9]+)")
+    set(CUDA_VERSION "${CMAKE_MATCH_1}")
   endif()
 endif()