remove CUDA

2026-03-21 22:12:47 +01:00
parent 6d9e6ca8ec
commit 91f6361055
2 changed files with 13 additions and 7 deletions
@@ -1,7 +1,14 @@
 FROM python:3.11-slim
+ENV PIP_NO_CACHE_DIR=1 \
+    PIP_DISABLE_PIP_VERSION_CHECK=1 \
+    PYTHONUNBUFFERED=1 \
+    TRANSFORMERS_NO_TF=1 \
+    HF_HUB_DISABLE_TELEMETRY=1
 WORKDIR /app
 COPY requirements.txt ./
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --upgrade pip setuptools wheel \
+    && pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.5.1 \
+    && pip install -r requirements.txt
 COPY . .
 EXPOSE 8001
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]