Docker Hub Page:
https://hub.docker.com/r/yanwk/sd-webui-base
Tags:
latest
: Recommended. Working well and nice performance.
nightly
: GitHub Actions auto builds. Plus dependencies for DreamBooth extension.
bulky
: Even more nightly version (PyTorch 2.1 + compile-installed xFormers, on CUDA 11.8) with devkit bundled, ready for compiling jobs in future. Building this image takes very long time, so I uploaded it manually.
Quick start:
mkdir -p storage
docker run -it \
--name sd-webui \
--gpus all \
-p 7860:7860 \
-v "$(pwd)"/storage:/home/runner \
--env CLI_ARGS="--xformers --medvram --allow-code --api --enable-insecure-extension-access" \
yanwk/sd-webui-base
I shall leave this repository for a while, and come back when PyTorch2 with CUDA 12.x is working.
Bonus: compiled wheel file of xFormers with PyTorch2.1 & TORCH_CUDA_ARCH_LIST=6.1;7.5;8.0;8.6. For anyone who googled to here :-)