diff --git a/cmake/ChooseCudaCC.cmake b/cmake/ChooseCudaCC.cmake index d9bc6c22..425e8bd5 100755 --- a/cmake/ChooseCudaCC.cmake +++ b/cmake/ChooseCudaCC.cmake @@ -80,8 +80,11 @@ function(chooseCudaCC SUPPORTED_CC SUPPORTED_GENCODE_FLAGS) # set(CUDA_MIN_CC 20) set(CUDA_MAX_CC 86) - if(CUDA_VERSION_MAJOR GREATER_EQUAL 11) + if(CUDA_VERSION VERSION_GREATER_EQUAL 11.1) set(CUDA_MIN_CC 35) + elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 11) + set(CUDA_MIN_CC 35) + set(CUDA_MAX_CC 80) elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 10) set(CUDA_MIN_CC 30) set(CUDA_MAX_CC 75)