A docker image for Stable Diffusion WebUI Forge or the AUTO1111 WebUI

Warning

This Dockerfile is based on on Cuda 12.4, which requires Nvidia driver >=545. In Ubuntu 22.04 you can update it by running ubuntu-drivers install nvidia:545 and a reboot. (Thanks @casao)

Docker compose

To run it, you will need the NVIDIA Container Toolkit

Create a directory. Inside this directory, create a subdirectory named data (if you don't and let docker create it, there might be permission issues) and a file named docker-compose.yml. Add the following content to the 'docker-compose.yml' file:

services:
  webui:
    image: "ghcr.io/yummiii/sd-webui-forge-docker:cuda-12.4.1"
    volumes:
      - "./data:/app/sd-webui"
    ports:
      - "7860:7860"
    environment:
      - "ARGS=--listen --enable-insecure-extension-access" # Insecure extension access is required if you want to install extensions with the listen flag
      - "UI=forge" # Specifies the UI that will be downloaded, forge for the forge webui or auto for the AUTOMATIC1111 webui
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

then run it with docker compose up.

You can set which arguments the webui will recive with the ARGS environment variable