|
@@ -18,8 +18,9 @@
|
|
|
#
|
|
#
|
|
|
|
|
|
|
|
if(CMAKE_CUDA_COMPILER_LOADED) # CUDA as a language
|
|
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()
|
|
|
endif()
|
|
endif()
|
|
|
|
|
|