Dockerfile

# build stage
FROM python:3.10 AS builder

# # # install PDM
RUN pip3 install -U pip setuptools wheel
RUN pip3 install pdm

# # copy files

COPY pyproject.toml pdm.lock README.md /app/

# # install dependencies and project into the local packages directory
WORKDIR /app
RUN mkdir __pypackages__ && pdm install --prod --no-lock --no-editable

FROM pytorch/pytorch

# retrieve packages from build stage
ENV PYTHONPATH=/app/pkgs

COPY --from=builder /app/__pypackages__/3.10/lib /app/pkgs
RUN pip uninstall -y accelerate
RUN pip install accelerate

#create project files
COPY run.sh pyproject.toml pdm.lock README.md /app/
COPY map_generation/ /app/map_generation
# COPY models/stable_unet_init/ /app/models/stable_unet_init
WORKDIR /app
RUN pip3 install -e .
ARG src="data/tiles"
# for local data
# COPY ${src} ./data 
VOLUME ./result_path
ENTRYPOINT ["/bin/bash", "run.sh"]