diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt index f71b438c..7b3f83c0 100644 --- a/cpp/CMakeLists.txt +++ b/cpp/CMakeLists.txt @@ -51,7 +51,10 @@ if(USE_BACKEND STREQUAL "CUDA") neuralnet/cudahelpers.cu ) # https://en.wikipedia.org/wiki/CUDA#GPUs_supported - if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.0) + if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 13.0) + set(CMAKE_CUDA_ARCHITECTURES 75 80 86 87 89 90) + set(CMAKE_CUDA_FLAGS "-Wno-deprecated-gpu-targets") + elseif(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.0) set(CMAKE_CUDA_ARCHITECTURES 50 52 53 60 61 62 70 72 75 80 86 87 90) set(CMAKE_CUDA_FLAGS "-Wno-deprecated-gpu-targets") elseif(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 11.8)