--- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -590,8 +590,14 @@ if(CUDA_FOUND) if (NOT DEFINED OSD_CUDA_NVCC_FLAGS) if (CUDA_VERSION_MAJOR LESS 6) set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_11 ) - else() + elseif (CUDA_VERSION_MAJOR LESS 9) set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_20 ) + elseif (CUDA_VERSION_MAJOR LESS 11) + set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_30 ) + elseif (CUDA_VERSION_MAJOR LESS 12) + set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_35 ) + else() + set( OSD_CUDA_NVCC_FLAGS --gpu-architecture compute_50 ) endif() endif() endif()