BLD: CUDA arch 86 not available until 11.1 (#543)

tomopy · Apr 29, 2021 · d04e73c · d04e73c
1 parent b06e891
commit d04e73c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmake/Modules/Packages.cmake b/cmake/Modules/Packages.cmake
@@ -212,7 +212,11 @@ if(TOMOPY_USE_CUDA)
             if(CUDA_VERSION_MAJOR VERSION_LESS 11)
                 set(CUDA_ARCH "30;32;35;37;50;52;53;60;61;62;70;72;75")
             else()
-                set(CUDA_ARCH "35;37;50;52;53;60;61;62;70;72;75;80;86")
+                if(CUDA_VERSION_MINOR VERSION_LESS 1)
+                    set(CUDA_ARCH "35;37;50;52;53;60;61;62;70;72;75;80")
+                else()
+                    set(CUDA_ARCH "35;37;50;52;53;60;61;62;70;72;75;80;86")
+                endif()
             endif()
         endif()