From 04eac7ededd6d92b614a941076c0f3070332d09b Mon Sep 17 00:00:00 2001 From: Antoine Jacquin Date: Sun, 31 May 2026 22:01:08 +0200 Subject: [PATCH] Support RTX 5060 Ti: CuPy 13.4 JIT compiles kernels at runtime for sm_120 --- lidar_pipeline/gpu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lidar_pipeline/gpu.py b/lidar_pipeline/gpu.py index 52aba97..2af3651 100644 --- a/lidar_pipeline/gpu.py +++ b/lidar_pipeline/gpu.py @@ -61,8 +61,8 @@ def _pick_gpu() -> int | None: _NUM_GPUS = len(gpus) gpus.sort(key=lambda g: g[4], reverse=True) - # CuPy 13.4 + CUDA 11.8 JIT supports up to sm_89 (RTX 40xx). - # sm_120 (RTX 50xx) is not supported yet — skip it. + # CuPy 13.4 + CUDA 11.8 JIT compiles kernels at runtime for any + # architecture (sm_89, sm_120, etc.). All NVIDIA GPUs are usable. for gpu in gpus: idx, name, cap_str, mem_mi, score, major = gpu if major <= 8: # sm_89 and below — works with CuPy 13.4 + JIT