mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-11 14:02:37 +08:00
89 lines
2.1 KiB
Docker
89 lines
2.1 KiB
Docker
# RunPod Serverless ComfyUI Worker
|
|
FROM runpod/pytorch:2.2.0-py3.11-cuda12.1.1-devel-ubuntu22.04
|
|
|
|
# Set working directory
|
|
WORKDIR /workspace
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
git \
|
|
wget \
|
|
curl \
|
|
unzip \
|
|
ffmpeg \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender-dev \
|
|
libglib2.0-0 \
|
|
libgl1-mesa-glx \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install Python dependencies
|
|
RUN pip install --no-cache-dir \
|
|
runpod \
|
|
requests \
|
|
pillow \
|
|
numpy \
|
|
torch \
|
|
torchvision \
|
|
torchaudio \
|
|
xformers \
|
|
accelerate \
|
|
transformers \
|
|
diffusers \
|
|
opencv-python \
|
|
scipy \
|
|
scikit-image
|
|
|
|
# Copy ComfyUI source code
|
|
COPY . /workspace/ComfyUI
|
|
|
|
# Set ComfyUI as working directory
|
|
WORKDIR /workspace/ComfyUI
|
|
|
|
# Install ComfyUI requirements
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
# Install additional dependencies for new features
|
|
RUN pip install --no-cache-dir \
|
|
safetensors \
|
|
transformers[torch] \
|
|
accelerate \
|
|
bitsandbytes \
|
|
optimum
|
|
|
|
# Copy environment example (users should provide their own .env)
|
|
COPY .env.example /workspace/.env.example
|
|
|
|
# Create necessary directories
|
|
RUN mkdir -p /workspace/ComfyUI/models/checkpoints \
|
|
/workspace/ComfyUI/models/vae \
|
|
/workspace/ComfyUI/models/loras \
|
|
/workspace/ComfyUI/models/controlnet \
|
|
/workspace/ComfyUI/models/clip_vision \
|
|
/workspace/ComfyUI/models/upscale_models \
|
|
/workspace/ComfyUI/input \
|
|
/workspace/ComfyUI/output \
|
|
/tmp/inputs \
|
|
/tmp/outputs \
|
|
/tmp/comfyui
|
|
|
|
# Set environment variables
|
|
ENV PYTHONPATH="/workspace/ComfyUI:${PYTHONPATH}"
|
|
ENV COMFYUI_SERVERLESS=true
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
|
|
# Create startup script
|
|
RUN echo '#!/bin/bash\n\
|
|
cd /workspace/ComfyUI\n\
|
|
python main.py --listen 0.0.0.0 --port 8000 --dont-print-server --disable-auto-launch &\n\
|
|
sleep 10\n\
|
|
cd /workspace/ComfyUI\n\
|
|
python runpod_handler.py' > /workspace/start.sh && chmod +x /workspace/start.sh
|
|
|
|
# Expose port
|
|
EXPOSE 8000
|
|
|
|
# Set the command
|
|
CMD ["/workspace/start.sh"] |