Add MIGraphX/half deps and use AMD onnxruntime wheel for ROCm 6.1.3
All checks were successful
Build ROCm Image / build (push) Successful in 8m1s

Per AMD docs (rocm.docs.amd.com install-onnx):
- apt install migraphx, migraphx-dev, half (required by onnxruntime-rocm)
- Switch to AMD-hosted wheel: onnxruntime_rocm-1.17.0-cp310 from repo.radeon.com
- Pin numpy==1.26.4 (numpy 2.0 incompatible with this wheel)
- Add MIGraphXExecutionProvider to provider list in engine.py

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-04-06 19:23:28 -04:00
parent e8507e0dea
commit f0ab3c1d59
4 changed files with 13 additions and 3 deletions

View File

@@ -1 +1,4 @@
onnxruntime-rocm
# AMD-provided onnxruntime wheel for ROCm 6.1.3 (Python 3.10)
# numpy must be pinned — 2.0 is incompatible with this wheel
https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/onnxruntime_rocm-1.17.0-cp310-cp310-linux_x86_64.whl
numpy==1.26.4