To use this container you would first need to install docker and docker-compose.
After that you would need deleted rocm/composable_kernel:ck_ub20.04_rocm5.5_rc4 docker image (alpha build of ROCM). You can download the image, using bittorect magnet link in the files of the gist. After downloading the image load it with docker load --input rocm5.5.tar.gz
Running is as simple as: sudo docker-compose up --build. After that Stable Diffusion Web UI could be accessed in http://127.0.0.1:3000
Many thanks to @wsippel for putting up step by step guide to get it working.
This is tested on kernel 6.2.10 and mesa 23.0.0
@rom1win
I actually don't use Docker much to run Stable Diffusion. This base image was built for testing when ROCm 5.5.1 released.
Based on your description of "unstable and crash" without specific logs, I tested it using this image and here are the steps (my host uses ROCm 5.6):
rocm_lab:rocm5.5.1-ub22.04-base
image with parameters like thisapt install git python3-pip python3-venv
to install missing dependenciesgit clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
to clone the a1111-webui repopip install -r requirements.txt
for a1111-webui's dependenciespip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.5
for torchexport HSA_OVERRIDE_GFX_VERSION=11.0.0
andexport HIP_VISIBLE_DEVICES=0
python3 launch.py --listen --enable-insecure-extension-access --opt-sdp-attention
to launchexport PYTORCH_HIP_ALLOC_CONF=garbage_collection_threshold:0.8,max_split_size_mb:512
and restart. Now 1024x1024 passes--opt-sdp-attention
from the arguments and restart. Now 1536x1536 passes, but it's slowI haven't observed any other crashes for now, and for generating large images, it is recommended to use Tiled VAE. If using Vlad's Automatic, the above optimizations are automatically enabled. Even without Tiled VAE, it can generate larger images. Don't use
--no-half-vae
.