Feature/rocm gpu detection - #140

Merged

solderzzc merged 7 commits into

feature/rocm-gpu-detection

Mar 9, 2026

Commits on Mar 8, 2026

fix(env_config): graceful CPU fallback when ROCm PyTorch unavailable
solderzzc
committed
eaefade

Commits on Mar 9, 2026

fix(env_config): proactive torch.cuda guard for ROCm PyTorch fallback
solderzzc
committed
fd87c7f
fix(deploy): prevent ultralytics from re-installing CPU onnxruntime
solderzzc
committed
dacf7cb
fix(deploy): simplify ROCm install — correct packages from the start
solderzzc
committed
bbd5db6
fix(deploy): auto-detect installed ROCm version for PyTorch index
solderzzc
committed
58f3d54
fix(deploy): fallback through ROCm versions for PyTorch wheels
solderzzc
committed
2d32d52
fix(rocm): use PyTorch+HIP for inference instead of ONNX
solderzzc
committed
28aede1