Files
rocm-chatterbox-whisper/Dockerfile.rocm
scott e8507e0dea
Some checks failed
Build ROCm Image / build (push) Has been cancelled
Install hipblas to satisfy onnxruntime-rocm provider dependency
libhipblas.so.2 is required by libonnxruntime_providers_rocm.so but was
missing from the base image install.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-04-06 19:20:45 -04:00

42 lines
1.0 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 \
&& 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"]