Update Dockerfile.cpu

This commit is contained in:
McCloudS
2025-02-03 23:29:06 -07:00
committed by GitHub
parent cb50d83464
commit d4a6a793af

View File

@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# Copy and install dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir --prefix=/install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install --no-cache-dir --prefix=/install -r torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu && pip install --no-cache-dir --prefix=/install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
# === Stage 2: Create a minimal runtime image ===
FROM python:3.11-slim-bullseye AS runtime