scott
f0ab3c1d59
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>
45 lines
1.1 KiB
Docker
45 lines
1.1 KiB
Docker
FROM rocm/dev-ubuntu-22.04:6.1
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONDONTWRITEBYTECODE=1 \
|
|
PYTHONUNBUFFERED=1 \
|
|
HF_HOME=/app/hf_cache \
|
|
PIP_NO_CACHE_DIR=1
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
python3 \
|
|
python3-pip \
|
|
python3-dev \
|
|
git \
|
|
ffmpeg \
|
|
libsndfile1 \
|
|
hipblas \
|
|
migraphx \
|
|
migraphx-dev \
|
|
half \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
|
|
# Step 1: Install onnxruntime-rocm first so it claims the onnxruntime namespace
|
|
# before any other package can pull in the CPU-only onnxruntime wheel.
|
|
COPY requirements-rocm-init.txt .
|
|
RUN pip3 install -r requirements-rocm-init.txt
|
|
|
|
# Step 2: Install remaining dependencies.
|
|
COPY requirements-rocm.txt .
|
|
RUN pip3 install -r requirements-rocm.txt
|
|
|
|
# Application source
|
|
COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./
|
|
|
|
# Default config (can be overridden by volume mount)
|
|
COPY config.yaml .
|
|
|
|
# Create default directories
|
|
RUN mkdir -p voices reference_audio hf_cache
|
|
|
|
EXPOSE 10200
|
|
|
|
CMD ["python3", "main.py"]
|