diff --git a/Dockerfile b/Dockerfile index 0dceb68..3a5005c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,6 +9,7 @@ FROM python:3.10-slim as dependencies RUN python3 -m venv /venv ARG PATH="/venv/bin:$PATH" RUN --mount=type=cache,target=/root/.cache/pip pip install --upgrade pip setuptools + # Install requirements RUN --mount=type=cache,target=/root/.cache/pip pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 diff --git a/Dockerfile.no_model b/Dockerfile.no_model index 8700411..a382dda 100644 --- a/Dockerfile.no_model +++ b/Dockerfile.no_model @@ -7,6 +7,7 @@ FROM python:3.10-slim as dependencies RUN python3 -m venv /venv ARG PATH="/venv/bin:$PATH" RUN --mount=type=cache,target=/root/.cache/pip pip install --upgrade pip setuptools + # Install requirements RUN --mount=type=cache,target=/root/.cache/pip pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 @@ -30,6 +31,11 @@ COPY --link --from=dependencies /venv /venv ARG PATH="/venv/bin:$PATH" ENV PATH=${PATH} +# Missing dependencies for arm64 +RUN if [ "$TARGETPLATFORM" = "linux/arm64" ]; then \ + apt-get install -y --no-install-recommends libgomp1 libsndfile1; \ + fi + # Non-root user RUN useradd -m -s /bin/bash appuser \ && mkdir -p /cache \