From 3f3f997163ec583b474824ffb9d180aaf07ee27b Mon Sep 17 00:00:00 2001 From: LambdaGo-lab <170946016+LambdaGo-lab@users.noreply.github.com> Date: Tue, 24 Feb 2026 20:40:14 +0900 Subject: [PATCH] Update cuda support gpu architecture --- cpp/CMakeLists.txt | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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)