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
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:
@@ -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
|
||||||
|
|||||||
@@ -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"]
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user