mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-01-30 20:33:06 -08:00
66 lines
1.9 KiB
Docker
66 lines
1.9 KiB
Docker
# Use an official CUDA runtime as the base image
|
|
FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
|
|
|
|
# Set environment variables
|
|
ENV WORKDIR /app
|
|
|
|
# Create a working directory
|
|
RUN mkdir -p $WORKDIR
|
|
WORKDIR $WORKDIR
|
|
|
|
# Create the outputs/img2img-images folder
|
|
RUN mkdir -p /home/ec2-user/apps/media_root
|
|
|
|
RUN apt-get update
|
|
|
|
# Install Python, pip, and other necessary dependencies
|
|
RUN apt-get update && \
|
|
apt-get install -y \
|
|
python3 \
|
|
python3-pip \
|
|
git \
|
|
wget \
|
|
libgl1 && \
|
|
ln -s /usr/bin/python3 /usr/bin/python && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Upgrade pip to the latest version
|
|
RUN pip install --upgrade pip
|
|
|
|
# Install the 'xformers' Python package
|
|
RUN pip install xformers
|
|
|
|
# Install additional system packages
|
|
RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6 libgl1 python3.10-venv --no-install-recommends google-perftools wget
|
|
|
|
# Clone the stable-diffusion-webui repository
|
|
RUN git clone https://github.com/Moonlite-Media/stable-diffusion-webui.git .
|
|
|
|
# Copy local files into the container
|
|
COPY . .
|
|
|
|
# Set up models folder and download required models
|
|
RUN mkdir -p models && \
|
|
cd models && \
|
|
mkdir -p Stable-diffusion && \
|
|
wget -q https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.safetensors
|
|
|
|
# Set up extensions folder and clone repositories
|
|
RUN mkdir -p extensions && \
|
|
cd extensions && \
|
|
git clone https://github.com/cheald/sd-webui-loractl.git && \
|
|
git clone https://github.com/Mikubill/sd-webui-controlnet && \
|
|
git clone https://github.com/deforum-art/sd-webui-deforum
|
|
|
|
# Install Python dependencies
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
# Install other external dependencies in a single command
|
|
RUN pip install --no-cache-dir python-dotenv insightface
|
|
|
|
# Run the application
|
|
CMD ["python", "launch.py", "--nowebui", "--deforum-api", "--listen", "--api", "--port", "7861"]
|
|
|
|
# Add fun fact
|
|
# Dogs are the best!
|