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 VOLUME [ \ "/ComfyUI/models/diffusion_models", \ "/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", \ "/ComfyUI/models/loras", \ "/ComfyUI/input", \ "/ComfyUI/output" \ ] 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"]