39 lines
1.1 KiB
Docker
39 lines
1.1 KiB
Docker
# BiRefNet background removal service — CUDA 12.4 runtime image.
|
|
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
UV_PYTHON_INSTALL_DIR=/opt/python \
|
|
UV_PROJECT_ENVIRONMENT=/app/.venv \
|
|
UV_COMPILE_BYTECODE=1 \
|
|
UV_LINK_MODE=copy \
|
|
HF_HOME=/app/hf_cache \
|
|
PORT=8000
|
|
|
|
# uv: fast, reproducible Python + dependency management.
|
|
COPY --from=ghcr.io/astral-sh/uv:0.9 /uv /uvx /bin/
|
|
|
|
RUN apt-get update \
|
|
&& apt-get install -y --no-install-recommends ca-certificates \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
|
|
# Install Python + dependencies first so this layer is cached across code changes.
|
|
# The BuildKit cache mount keeps the uv download cache warm across rebuilds.
|
|
COPY pyproject.toml ./
|
|
RUN --mount=type=cache,target=/root/.cache/uv \
|
|
uv python install 3.12 \
|
|
&& uv sync --no-install-project --no-dev
|
|
|
|
# Application code.
|
|
COPY src ./src
|
|
COPY README.md ./
|
|
RUN --mount=type=cache,target=/root/.cache/uv \
|
|
uv sync --no-dev
|
|
|
|
ENV PATH="/app/.venv/bin:${PATH}"
|
|
|
|
EXPOSE 8000
|
|
CMD ["birefnet-service"]
|