From 5df9321890f55f108d7fd283560d8997783f56f3 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 12 Feb 2025 04:29:09 +0300 Subject: [PATCH] Add ipex dockerfile --- configs/Dockerfile.ipex | 99 ++++++++++++++++++++++++++++++++++++++++- 1 file changed, 98 insertions(+), 1 deletion(-) diff --git a/configs/Dockerfile.ipex b/configs/Dockerfile.ipex index 38362d11f..edeb9dbdf 100644 --- a/configs/Dockerfile.ipex +++ b/configs/Dockerfile.ipex @@ -1,2 +1,99 @@ -# SD.Next Dockerfile +# SD.Next IPEX Dockerfile # docs: + +# base image +FROM ubuntu:noble + +# metadata +LABEL org.opencontainers.image.vendor="SD.Next" +LABEL org.opencontainers.image.authors="disty0" +LABEL org.opencontainers.image.url="https://github.com/vladmandic/sdnext/" +LABEL org.opencontainers.image.documentation="https://github.com/vladmandic/sdnext/wiki/Docker" +LABEL org.opencontainers.image.source="https://github.com/vladmandic/sdnext/" +LABEL org.opencontainers.image.licenses="AGPL-3.0" +LABEL org.opencontainers.image.title="SD.Next IPEX" +LABEL org.opencontainers.image.description="SD.Next: Advanced Implementation of Stable Diffusion and other Diffusion-based generative image models" +LABEL org.opencontainers.image.base.name="https://hub.docker.com/_/ubuntu:noble" +LABEL org.opencontainers.image.version="latest" + +# essentials +RUN apt-get update && \ + apt-get install -y --no-install-recommends --fix-missing \ + software-properties-common \ + build-essential \ + ca-certificates \ + wget \ + gpg \ + git + + +# intel compute runtime +RUN wget -qO - https://repositories.intel.com/gpu/intel-graphics.key | gpg --yes --dearmor --output /usr/share/keyrings/intel-graphics.gpg +RUN echo "deb [arch=amd64,i386 signed-by=/usr/share/keyrings/intel-graphics.gpg] https://repositories.intel.com/gpu/ubuntu noble client" | tee /etc/apt/sources.list.d/intel-gpu-noble.list +RUN apt-get update + +RUN apt-get install -y --no-install-recommends --fix-missing \ + intel-opencl-icd \ + libze-intel-gpu1 \ + libze1 + +# required by pytorch / ipex +RUN apt-get install -y --no-install-recommends --fix-missing \ + libgl1 \ + libglib2.0-0 \ + libgomp1 + +# python3.12 +RUN apt-get install -y --no-install-recommends --fix-missing \ + python3 \ + python3-dev \ + python3-venv \ + python3-pip + +# jemalloc is not required but it is highly recommended (also used with optional ipexrun) +RUN apt-get install -y --no-install-recommends --fix-missing libjemalloc-dev +ENV LD_PRELOAD=libjemalloc.so.2 + +# cleanup +RUN /usr/sbin/ldconfig +RUN apt-get clean && rm -rf /var/lib/apt/lists/* + +# stop pip and uv from caching +ENV PIP_NO_CACHE_DIR=true +ENV UV_NO_CACHE=true + +# set paths to use with sdnext +ENV SD_DOCKER=true +ENV SD_DATADIR="/mnt/data" +ENV SD_MODELSDIR="/mnt/models" +ENV venv_dir="/mnt/python/venv" + +# paths used by sdnext can be a volume if necessary +#VOLUME [ "/app" ] +#VOLUME [ "/mnt/data" ] +#VOLUME [ "/mnt/models" ] +#VOLUME [ "/mnt/python" ] +#VOLUME [ "/root/.cache/huggingface" ] + +# intel specific environment variables +#ENV IPEX_SDPA_SLICE_TRIGGER_RATE=1 +#ENV IPEX_ATTENTION_SLICE_RATE=0.5 +#ENV IPEX_FORCE_ATTENTION_SLICE=-1 +#ENV IPEXRUN=False + +# git clone and run sdnext +RUN echo '#!/bin/bash\ngit status || git clone https://github.com/vladmandic/sdnext.git .\n/app/webui.sh "$@"' | tee /bin/startup.sh +RUN chmod 755 /bin/startup.sh + +# actually run sdnext +WORKDIR /app +ENTRYPOINT [ "startup.sh", "-f", "--use-ipex", "--listen", "--debug", "--api-log", "--log", "sdnext.log" ] + +# expose port +EXPOSE 7860 + +# healthcheck function +# HEALTHCHECK --interval=60s --timeout=10s --start-period=60s --retries=3 CMD curl --fail http://localhost:7860/sdapi/v1/status || exit 1 + +# stop signal +STOPSIGNAL SIGINT