Dockerfile at main - qte77/SegFormerQuantization

main

SegFormerQuantization

List view

Dockerfile

0 Duplication
9 Churn
45 Lines
33 Lines of Code

ARG APP_ROOT="/src"
ARG PYTHON_VERSION="3.12"
ARG USER="appuser"


# Stage 1: Builder Image
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel AS builder
LABEL author="qte77"
LABEL builder=true
ENV PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1
COPY pyproject.toml uv.lock ./
RUN set -xe \
    && pip install --no-cache-dir uv \
    && uv sync --frozen


# Stage 2: Runtime Image
FROM python:${PYTHON_VERSION}-slim AS runtime
LABEL author="qte77"
LABEL runtime=true

ARG APP_ROOT
ARG USER
ENV PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PYTHONPATH=${APP_ROOT} \
    PATH="${APP_ROOT}:${PATH}"
#    WANDB_KEY=${WANDB_KEY} \
#    WANDB_DISABLE_CODE=true

RUN set -xe \
    && useradd --create-home ${USER} \
    && pip install --no-cache-dir uv
    
USER ${USER}
WORKDIR ${APP_ROOT}
COPY --from=builder /.venv .venv
COPY --chown=${USER}:${USER} ${APP_ROOT} .

CMD [ \
    "uv", "run", \
    "--locked", "--no-sync", \
    "python", "-m", "." \
]