From a5af50c04335987783b98c780302826b1897c5cc Mon Sep 17 00:00:00 2001 From: Antoine Jacquin Date: Sun, 31 May 2026 22:04:20 +0200 Subject: [PATCH] Remove sm_120 filter: CuPy 13.4 JIT supports RTX 5060 Ti --- lidar_pipeline/gpu.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/lidar_pipeline/gpu.py b/lidar_pipeline/gpu.py index 2af3651..1b1f50f 100644 --- a/lidar_pipeline/gpu.py +++ b/lidar_pipeline/gpu.py @@ -65,15 +65,11 @@ def _pick_gpu() -> int | None: # 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 - _best_gpu_id = idx - _gpu_name = name - _gpu_mem_gb = mem_mi // 1024 - HAS_GPU = True - return _best_gpu_id - - # All GPUs have unsupported compute capability - _gpu_reason = "aucun GPU avec compute capability compatible CuPy 13.4" + _best_gpu_id = idx + _gpu_name = name + _gpu_mem_gb = mem_mi // 1024 + HAS_GPU = True + return _best_gpu_id return None except (FileNotFoundError, subprocess.TimeoutExpired, Exception): @@ -177,7 +173,7 @@ def log_gpu_status(): gpu_info = f"GPU: {_gpu_name} ({_gpu_mem_gb} Go VRAM)" logger.info(gpu_info) - # Warn about unsupported GPUs that exist but are not used + # List other GPUs for info try: import subprocess result = subprocess.run( @@ -192,7 +188,7 @@ def log_gpu_status(): idx = int(parts[0]) if idx != _best_gpu_id: cap = parts[2] - logger.info(f" GPU {idx}: {parts[1]} (sm_{cap}) — non utilisé (incompatible CuPy)") + logger.info(f" GPU {idx}: {parts[1]} (sm_{cap}) — disponible") except Exception: pass else: