Remove sm_120 filter: CuPy 13.4 JIT supports RTX 5060 Ti
This commit is contained in:
@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user