Merge pull request #116 from alicevision/fix/cuda11

[cmake] sm 86 is only for cuda >= 11.1
alicevision · Oct 29, 2020 · fafcad9 · fafcad9
2 parents c8a669b + d6cb770
commit fafcad9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)