remove CUDA
This commit is contained in:
@@ -1,7 +1,14 @@
|
|||||||
FROM python:3.11-slim
|
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
|
WORKDIR /app
|
||||||
COPY requirements.txt ./
|
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 . .
|
COPY . .
|
||||||
EXPOSE 8001
|
EXPOSE 8001
|
||||||
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]
|
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
fastapi>=0.85
|
fastapi==0.115.12
|
||||||
uvicorn[standard]>=0.18
|
uvicorn[standard]==0.34.0
|
||||||
transformers>=4.30
|
transformers==4.48.3
|
||||||
torch>=1.13
|
cachetools==5.5.2
|
||||||
cachetools>=5.0
|
pydantic==2.10.6
|
||||||
pydantic>=1.10
|
|
||||||
|
|||||||
Reference in New Issue
Block a user