Dockerfile

FROM nvcr.io/nvidia/pytorch:23.04-py3 as base
ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/London

RUN apt-get update && \
    apt-get install -y git curl python3-venv python3-tk libgl1 libglib2.0-0 libgoogle-perftools-dev && \
    rm -rf /var/lib/apt/lists/*

RUN useradd -m -s /bin/bash appuser
USER appuser

WORKDIR /app

RUN python3 -m venv ./venv && . ./venv/bin/activate && \
    python3 -m pip install wheel

COPY requirements.txt setup.py .
RUN . ./venv/bin/activate && \
    python3 -m pip install --use-pep517 -U -r requirements.txt triton torch>=2.0.0+cu121 xformers \
                           --extra-index-url https://download.pytorch.org/whl/cu121

USER appuser
COPY --chown=appuser . .

STOPSIGNAL SIGINT
ENV LD_PRELOAD=libtcmalloc.so
CMD . ./venv/bin/activate && ./gui.sh --listen 0.0.0.0 --server_port 7860