mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2026-01-05 08:09:59 +01:00
41 lines
1.2 KiB
Docker
41 lines
1.2 KiB
Docker
FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
|
|
|
|
RUN apt-get update && apt-get install -y git && apt-get clean
|
|
|
|
# add some packages for some custom nodes in comfyui
|
|
RUN apt-get install 'libglib2.0-0' -y
|
|
RUN apt-get update
|
|
RUN apt-get install 'libgl1-mesa-glx' -y
|
|
RUN apt-get install 'python-dev' -y
|
|
|
|
# install pip package for codeformer face detection custom node
|
|
RUN pip install lpips
|
|
|
|
ENV ROOT=/stable-diffusion
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
|
|
cd ${ROOT} && \
|
|
git checkout master && \
|
|
git reset --hard d1f3637a5a944d0607b899babd8ff11d87100503 && \
|
|
pip install -r requirements.txt
|
|
|
|
WORKDIR ${ROOT}
|
|
|
|
ARG BRANCH=master SHA=c97be4db91d4a249c19afdf88fa1cf3268544e45
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
git fetch && \
|
|
git checkout ${BRANCH} && \
|
|
git reset --hard ${SHA} && \
|
|
pip install -r requirements.txt
|
|
|
|
# add info
|
|
COPY . /docker/
|
|
RUN chmod u+x /docker/entrypoint.sh && cp /docker/extra_model_paths.yaml ${ROOT}
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES=all PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
|
|
EXPOSE 7860
|
|
ENTRYPOINT ["/docker/entrypoint.sh"]
|
|
CMD python -u main.py --listen --port 7860 ${CLI_ARGS}
|