Update Dockerfile.cpu

2025-02-04 00:00:33 -07:00
parent a0e79f27bd
commit 16b5e63adf
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 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
+RUN pip install --no-cache-dir --prefix=/install -r requirements.txt

 # === Stage 2: Create a minimal runtime image ===
 FROM python:3.11-slim-bullseye AS runtime