49 lines
1.2 KiB
Docker
49 lines
1.2 KiB
Docker
|
# RunPod AI Orchestrator Template
|
||
|
|
# Minimal Docker image for ComfyUI + vLLM orchestration
|
||
|
|
# Models and application code live on network volume at /workspace
|
||
|
|
|
||
|
|
FROM runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04
|
||
|
|
|
||
|
|
# Install additional system utilities
|
||
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
||
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||
|
|
build-essential \
|
||
|
|
python3-dev \
|
||
|
|
python3-pip \
|
||
|
|
python3-venv \
|
||
|
|
git \
|
||
|
|
curl \
|
||
|
|
wget \
|
||
|
|
htop \
|
||
|
|
tmux \
|
||
|
|
net-tools \
|
||
|
|
ffmpeg \
|
||
|
|
libavcodec-dev \
|
||
|
|
libavformat-dev \
|
||
|
|
libavutil-dev \
|
||
|
|
libswscale-dev \
|
||
|
|
bc \
|
||
|
|
jq \
|
||
|
|
openssh-server \
|
||
|
|
&& apt-get clean && rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
# Install Supervisor for process management
|
||
|
|
RUN pip install --upgrade pip && pip install --no-cache-dir supervisor
|
||
|
|
|
||
|
|
# Install Tailscale for VPN connectivity
|
||
|
|
RUN curl -fsSL https://tailscale.com/install.sh | sh
|
||
|
|
|
||
|
|
# Install yq
|
||
|
|
RUN wget https://github.com/mikefarah/yq/releases/latest/download/yq_linux_amd64 -O /usr/local/bin/yq && \
|
||
|
|
chmod +x /usr/local/bin/yq
|
||
|
|
|
||
|
|
# Copy the startup script
|
||
|
|
COPY start.sh /start.sh
|
||
|
|
RUN chmod +x /start.sh
|
||
|
|
|
||
|
|
# Set working directory to /workspace (network volume mount point)
|
||
|
|
WORKDIR /workspace
|
||
|
|
|
||
|
|
# RunPod calls /start.sh by default
|
||
|
|
CMD ["/start.sh"]
|