example / Dockerfile
jessehostetler's picture
Default to CPU-only version of torch. Improve docs.
c2feb3e
raw
history blame contribute delete
562 Bytes
FROM python:3.12-slim
ENV PYTHONDONTWRITEBYTECODE="1" \
PYTHONUNBUFFERED="1"
# hadolint ignore=DL3013
RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
WORKDIR /app/
COPY requirements.cpu.txt ./
RUN python3 -m pip install --no-cache-dir -r ./requirements.cpu.txt
COPY requirements.torch.cpu.txt ./
RUN python3 -m pip install --no-cache-dir -r ./requirements.torch.cpu.txt
COPY app ./app
COPY models ./models
COPY main.py ./
EXPOSE 8000
ENTRYPOINT ["python3", "-m", "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]