6
6
endif ()
7
7
8
8
set (CUDA_FIND_QUIETLY TRUE )
9
- find_package (CUDA 7.5 )
9
+ find_package (CUDA 9.0 )
10
10
11
11
if (CUDA_FOUND )
12
12
message (STATUS "Found CUDA Toolkit v${CUDA_VERSION_STRING} " )
@@ -23,17 +23,19 @@ if(CUDA_FOUND)
23
23
# To add support of older GPU for kinfu, I would embed PTX 11 and 12 into so-file. GPU with sm_13 will run PTX 12 code (no difference for kinfu)
24
24
25
25
# Find a complete list for CUDA compute capabilities at http://developer.nvidia.com/cuda-gpus
26
-
27
- if (NOT ${CUDA_VERSION_STRING} VERSION_LESS "10.0" )
26
+
27
+ # For a list showing CUDA toolkit version support for compute capabilities see: https://en.wikipedia.org/wiki/CUDA
28
+ # or the nvidia release notes ie:
29
+ # https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-general-new-features
30
+ # or
31
+ # https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#deprecated-features
32
+
33
+ if (NOT ${CUDA_VERSION_STRING} VERSION_LESS "11.0" )
34
+ set (__cuda_arch_bin "5.2 5.3 6.0 6.1 7.0 7.2 7.5" )
35
+ elseif (NOT ${CUDA_VERSION_STRING} VERSION_LESS "10.0" )
28
36
set (__cuda_arch_bin "3.0 3.5 5.0 5.2 5.3 6.0 6.1 7.0 7.2 7.5" )
29
- elseif (NOT ${CUDA_VERSION_STRING} VERSION_LESS "9.1" )
30
- set (__cuda_arch_bin "3.0 3.5 5.0 5.2 5.3 6.0 6.1 7.0 7.2" )
31
37
elseif (NOT ${CUDA_VERSION_STRING} VERSION_LESS "9.0" )
32
- set (__cuda_arch_bin "3.0 3.5 5.0 5.2 5.3 6.0 6.1 7.0" )
33
- elseif (NOT ${CUDA_VERSION_STRING} VERSION_LESS "8.0" )
34
- set (__cuda_arch_bin "2.0 2.1(2.0) 3.0 3.5 5.0 5.2 5.3 6.0 6.1" )
35
- else ()
36
- set (__cuda_arch_bin "2.0 2.1(2.0) 3.0 3.5 5.0 5.2" )
38
+ set (__cuda_arch_bin "3.0 3.5 5.0 5.2 5.3 6.0 6.1 7.0 7.2" )
37
39
endif ()
38
40
39
41
set (CUDA_ARCH_BIN ${__cuda_arch_bin} CACHE STRING "Specify 'real' GPU architectures to build binaries for, BIN(PTX) format is supported" )
0 commit comments