Dockerfile

FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04

SHELL ["/bin/bash", "-o", "pipefail", "-c"]

ENV SHELL=/bin/bash

# Prevents prompts from packages asking for user input during installation
ENV DEBIAN_FRONTEND=noninteractive
# Prefer binary wheels over source distributions for faster pip installations
ENV PIP_PREFER_BINARY=1
# Ensures output from python is printed immediately to the terminal without buffering
ENV PYTHONUNBUFFERED=1 

WORKDIR /

RUN apt-get update --yes && \
	apt-get upgrade --yes && \
	apt install --yes --no-install-recommends \
	python3.10 python3.10-venv python3-pip \
	git libgl1 nodejs rsync wget

RUN apt-get autoremove -y && apt-get clean && rm -rf /var/lib/apt/lists/* 

RUN pip install -U comfy-cli

RUN /usr/bin/yes | comfy --workspace /comfyui install --cuda-version 12.1 --nvidia --version latest 

COPY workflow.json /workflow.json
WORKDIR /comfyui
RUN comfy node install-deps --workflow=/workflow.json

COPY models/ /comfyui/models/

WORKDIR /

COPY --chmod=755 start.sh /

SHELL ["/bin/bash", "--login", "-c"]
CMD ["/start.sh"]