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
ENV TRANSFORMERS_CACHE=/tmp/hf-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 all directories and set full write permissions
RUN mkdir -p /app/outputs /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 /tmp/hf-cache && \
    touch /app/config_modification_tutorial.txt && \
    chmod -R 777 /app /tmp/matplotlib-cache /tmp/numba-cache /tmp/hf-cache

EXPOSE 7860

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