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

@@ -14,6 +14,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
ffmpeg \ ffmpeg \
libsndfile1 \ libsndfile1 \
hipblas \ hipblas \
migraphx \
migraphx-dev \
half \
&& rm -rf /var/lib/apt/lists/* && rm -rf /var/lib/apt/lists/*
WORKDIR /app WORKDIR /app

View File

@@ -37,7 +37,11 @@ def detect_device() -> str:
def _get_providers(device: str) -> list: def _get_providers(device: str) -> list:
if device in ("rocm", "cuda"): if device in ("rocm", "cuda"):
return [("ROCMExecutionProvider", {"device_id": 0}), "CPUExecutionProvider"] return [
"MIGraphXExecutionProvider",
("ROCMExecutionProvider", {"device_id": 0}),
"CPUExecutionProvider",
]
return ["CPUExecutionProvider"] return ["CPUExecutionProvider"]

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

View File

@@ -1,5 +1,5 @@
# Audio processing # Audio processing
numpy>=1.24.0,<2.0.0 numpy==1.26.4
soundfile soundfile
librosa==0.11.0 librosa==0.11.0