mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2026-01-12 19:49:57 +01:00
30 lines
838 B
Docker
30 lines
838 B
Docker
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
ENV PIP_PREFER_BINARY=1
|
|
ENV APPLICATION_ROOT=/stable-diffusion
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
ENV PYTHONPATH="${PYTHONPATH}:${PWD}"
|
|
ENV CLI_ARGS=""
|
|
|
|
WORKDIR ${APPLICATION_ROOT}
|
|
|
|
# Explore how we can use the CLI to do this instead
|
|
# https://docs.comfy.org/installation/system_requirements
|
|
# https://docs.comfy.org/comfy-cli/getting-started
|
|
RUN apt-get update && \
|
|
apt-get install -y git && \
|
|
apt-get clean && \
|
|
git clone --branch v0.3.19 --depth 1 https://github.com/comfyanonymous/ComfyUI.git ${APPLICATION_ROOT}
|
|
|
|
COPY . .
|
|
|
|
RUN --mount=type=cache,target=/root/.cache/pip \
|
|
pip install -r requirements.txt && \
|
|
chmod u+x ./entrypoint.sh
|
|
|
|
EXPOSE 7860
|
|
|
|
ENTRYPOINT ["./entrypoint.sh"]
|
|
CMD python -u main.py --listen --port 7860 ${CLI_ARGS}
|