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: