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-baseimage with parameters like thisapt install git python3-pip python3-venvto install missing dependenciesgit clone https://github.com/AUTOMATIC1111/stable-diffusion-webuito clone the a1111-webui repopip install -r requirements.txtfor a1111-webui's dependenciespip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.5for torchexport HSA_OVERRIDE_GFX_VERSION=11.0.0andexport HIP_VISIBLE_DEVICES=0python3 launch.py --listen --enable-insecure-extension-access --opt-sdp-attentionto launchexport PYTORCH_HIP_ALLOC_CONF=garbage_collection_threshold:0.8,max_split_size_mb:512and restart. Now 1024x1024 passes--opt-sdp-attentionfrom 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.