Update Dockerfile.cpu

2025-02-04 00:20:55 -07:00
parent f12d9e258f
commit 7a6bd076c1
1 changed files with 1 additions and 1 deletions
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -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
+RUN pip install --no-cache-dir --prefix=/install 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