Remove sm_120 filter: CuPy 13.4 JIT supports RTX 5060 Ti

This commit is contained in:
Antoine Jacquin
2026-05-31 22:04:20 +02:00
parent 04eac7eded
commit a5af50c043

View File

@ -65,15 +65,11 @@ def _pick_gpu() -> int | None:
# architecture (sm_89, sm_120, etc.). All NVIDIA GPUs are usable. # architecture (sm_89, sm_120, etc.). All NVIDIA GPUs are usable.
for gpu in gpus: for gpu in gpus:
idx, name, cap_str, mem_mi, score, major = gpu 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
_best_gpu_id = idx _gpu_name = name
_gpu_name = name _gpu_mem_gb = mem_mi // 1024
_gpu_mem_gb = mem_mi // 1024 HAS_GPU = True
HAS_GPU = True return _best_gpu_id
return _best_gpu_id
# All GPUs have unsupported compute capability
_gpu_reason = "aucun GPU avec compute capability compatible CuPy 13.4"
return None return None
except (FileNotFoundError, subprocess.TimeoutExpired, Exception): except (FileNotFoundError, subprocess.TimeoutExpired, Exception):
@ -177,7 +173,7 @@ def log_gpu_status():
gpu_info = f"GPU: {_gpu_name} ({_gpu_mem_gb} Go VRAM)" gpu_info = f"GPU: {_gpu_name} ({_gpu_mem_gb} Go VRAM)"
logger.info(gpu_info) logger.info(gpu_info)
# Warn about unsupported GPUs that exist but are not used # List other GPUs for info
try: try:
import subprocess import subprocess
result = subprocess.run( result = subprocess.run(
@ -192,7 +188,7 @@ def log_gpu_status():
idx = int(parts[0]) idx = int(parts[0])
if idx != _best_gpu_id: if idx != _best_gpu_id:
cap = parts[2] 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: except Exception:
pass pass
else: else: