ARG RELEASE |
ARG LAUNCHPAD_BUILD_ARCH |
LABEL org.opencontainers.image.ref.name=ubuntu |
LABEL org.opencontainers.image.version=20.04 |
ADD file:9169bb1d6ef21313aed17e924538fee03d858460ae6b05e01968457dfc043bd7 in / |
CMD ["/bin/bash"] |
ARG PYTORCH_VERSION |
ARG TRITON_VERSION |
ARG TARGETPLATFORM |
ARG CUDA_VERSION |
LABEL com.nvidia.volumes.needed=nvidia_driver |
RUN |4 PYTORCH_VERSION=2.1.2 TRITON_VERSION=2.1.0+e6216047b8 TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.1 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit |
COPY /opt/conda /opt/conda # buildkit |
RUN |4 PYTORCH_VERSION=2.1.2 TRITON_VERSION=2.1.0+e6216047b8 TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.1 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit |
ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
ENV NVIDIA_VISIBLE_DEVICES=all |
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
ENV PYTORCH_VERSION=2.1.2 |
WORKDIR /workspace |
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1 |
RUN /bin/sh -c apt-get update && apt-get install -y fonts-dejavu-core rsync git jq moreutils aria2 ffmpeg libglfw3-dev libgles2-mesa-dev pkg-config libcairo2 libcairo2-dev build-essential # buildkit |
WORKDIR / |
RUN /bin/sh -c git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && cd stable-diffusion-webui && git reset --hard cf2772fab0af5573da775e7437e6acdca424f26e && pip install -r requirements_versions.txt # buildkit |
ENV ROOT=/stable-diffusion-webui |
COPY /repositories/ /stable-diffusion-webui/repositories/ # buildkit |
RUN /bin/sh -c mkdir ${ROOT}/interrogate && cp ${ROOT}/repositories/clip-interrogator/clip_interrogator/data/* ${ROOT}/interrogate # buildkit |
RUN /bin/sh -c pip install -r ${ROOT}/repositories/CodeFormer/requirements.txt # buildkit |
RUN /bin/sh -c pip install pyngrok xformers==0.0.23.post1 git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 git+https://github.com/mlfoundations/open_clip.git@v2.20.0 # buildkit |
RUN /bin/sh -c apt-get -y install libgoogle-perftools-dev && apt-get clean # buildkit |
ENV LD_PRELOAD=libtcmalloc.so |
COPY . /docker # buildkit |
RUN /bin/sh -c sed -i 's/in_app_dir = .*/in_app_dir = True/g' /opt/conda/lib/python3.10/site-packages/gradio/routes.py && git config --global --add safe.directory '*' # buildkit |
WORKDIR /stable-diffusion-webui |
ENV NVIDIA_VISIBLE_DEVICES=all |
ENV CLI_ARGS= |
EXPOSE map[7860/tcp:{}] |
ENTRYPOINT ["/docker/entrypoint.sh"] |
CMD ["/bin/sh" "-c" "python -u webui.py --listen --port 7860 ${CLI_ARGS}"] |