Skip to content

Commit fafcad9

Browse files
authored
Merge pull request #116 from alicevision/fix/cuda11
[cmake] sm 86 is only for cuda >= 11.1
2 parents c8a669b + d6cb770 commit fafcad9

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

cmake/ChooseCudaCC.cmake

+4-1
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,11 @@ function(chooseCudaCC SUPPORTED_CC SUPPORTED_GENCODE_FLAGS)
8080
#
8181
set(CUDA_MIN_CC 20)
8282
set(CUDA_MAX_CC 86)
83-
if(CUDA_VERSION_MAJOR GREATER_EQUAL 11)
83+
if(CUDA_VERSION VERSION_GREATER_EQUAL 11.1)
8484
set(CUDA_MIN_CC 35)
85+
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 11)
86+
set(CUDA_MIN_CC 35)
87+
set(CUDA_MAX_CC 80)
8588
elseif(CUDA_VERSION_MAJOR GREATER_EQUAL 10)
8689
set(CUDA_MIN_CC 30)
8790
set(CUDA_MAX_CC 75)

0 commit comments

Comments
 (0)