Dockerfile

FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime

ENV TZ=Asia/Shanghai
ENV MPLCONFIGDIR=/tmp/matplotlib-cache
ENV NUMBA_CACHE_DIR=/tmp/numba-cache

WORKDIR /app

COPY . /app

RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir opencv-python-headless -i https://pypi.org/simple

# Create directories and set permissions
RUN mkdir -p /app/presets /app/logs /app/repositories/Fooocus/models/sam /app/repositories/Fooocus/models/checkpoints /app/repositories/Fooocus/models/loras /app/repositories/Fooocus/models/embeddings /app/repositories/Fooocus/models/vae_approx /app/repositories/Fooocus/models/upscale_models /app/repositories/Fooocus/models/inpaint /app/repositories/Fooocus/models/controlnet /app/repositories/Fooocus/outputs /tmp/matplotlib-cache /tmp/numba-cache && \
    touch /app/config_modification_tutorial.txt && \
    chmod -R 777 /app/presets /app/logs /app/repositories/Fooocus /app/config_modification_tutorial.txt /app/config.txt /tmp/matplotlib-cache /tmp/numba-cache

EXPOSE 8888

CMD ["python", "main.py", "--host", "0.0.0.0", "--port", "8888", "--skip-pip"]