code_public/lidar_rendu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove sm_120 filter: CuPy 13.4 JIT supports RTX 5060 Ti

Browse Source
This commit is contained in:
Antoine Jacquin
2026-05-31 22:04:20 +02:00
parent 04eac7eded
commit a5af50c043
1 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lidar_pipeline/gpu.py
View File

@ -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: