mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-07-06 07:17:09 +08:00
46 lines
1.5 KiB
Docker
46 lines
1.5 KiB
Docker
FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04 AS base
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# Install necessary dependencies and Python 3.12
|
|
RUN apt-get update \
|
|
&& apt-get install -y \
|
|
git \
|
|
software-properties-common \
|
|
curl \
|
|
python3.12 \
|
|
python3.12-dev \
|
|
python3.12-venv \
|
|
python3-setuptools \
|
|
wget \
|
|
ffmpeg \
|
|
libsm6 \
|
|
libxext6 \
|
|
libgl1 \
|
|
&& apt-get clean \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Set the working directory
|
|
WORKDIR /app
|
|
|
|
# Clone the ComfyUI repository and set up virtual environment
|
|
RUN git clone https://github.com/comfyanonymous/ComfyUI.git /app/comfyui \
|
|
&& python3.12 -m venv /app/venv \
|
|
&& /app/venv/bin/pip install --upgrade pip \
|
|
&& /app/venv/bin/pip install pyyaml \
|
|
&& /app/venv/bin/pip install -r /app/comfyui/requirements.txt
|
|
|
|
# Install PyTorch with CUDA 12.6 support (stable version)
|
|
RUN /app/venv/bin/pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126
|
|
|
|
# Clone ComfyUI-Manager and install its dependencies
|
|
RUN git clone https://github.com/ltdrdata/ComfyUI-Manager.git /app/temp/ComfyUI-Manager \
|
|
&& mv /app/temp/* /app/comfyui/custom_nodes/ \
|
|
&& rm -rf /app/temp \
|
|
&& /app/venv/bin/pip install -r /app/comfyui/custom_nodes/ComfyUI-Manager/requirements.txt
|
|
|
|
# Expose the backend port
|
|
EXPOSE 8188
|
|
|
|
# Set the entrypoint to run the app
|
|
CMD ["/bin/bash", "-c", "source /app/venv/bin/activate && python3 /app/comfyui/main.py --listen 0.0.0.0 --port 8188"]
|