ComfyUI-docker/Dockerfile

43 lines
1.1 KiB
Docker
Raw Normal View History

2024-10-12 23:27:14 +00:00
FROM nvidia/cuda:12.6.1-cudnn-runtime-ubuntu20.04
RUN apt-get update && apt-get install -y \
git \
python3 \
python3-pip \
build-essential \
&& rm -rf /var/lib/apt/lists/*
RUN git clone https://github.com/comfyanonymous/ComfyUI.git /ComfyUI
WORKDIR /ComfyUI
2024-10-12 23:28:25 +00:00
VOLUME [ \
"/ComfyUI/models/diffusion_models", \
2024-10-12 23:27:14 +00:00
"/ComfyUI/models/controlnet", \
"/ComfyUI/models/upscale_models", \
"/ComfyUI/models/photomaker", \
"/ComfyUI/models/embeddings", \
"/ComfyUI/models/checkpoints", \
"/ComfyUI/models/style_models", \
"/ComfyUI/models/clip", \
"/ComfyUI/models/hypernetworks", \
"/ComfyUI/models/diffusers", \
"/ComfyUI/models/vae", \
"/ComfyUI/models/gligen", \
"/ComfyUI/models/unet", \
"/ComfyUI/models/clip_vision", \
"/ComfyUI/models/vae_approx", \
2024-10-14 22:44:17 +00:00
"/ComfyUI/models/loras", \
"/ComfyUI/input", \
"/ComfyUI/output" \
2024-10-12 23:27:14 +00:00
]
RUN pip3 install --upgrade pip \
&& pip3 install -r requirements.txt
ENV PATH /usr/local/cuda/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:${LD_LIBRARY_PATH}
CMD ["python3", "/ComfyUI/main.py", "--listen"]