Images for stable-diffusion-webui.

In particular this repository contains support for AMD GPUs that have ROCm enabled.

Example usage

For AMD cards, you need to install at the very least the DKMS driver⁠ with amdgpu-install --usecase=dkms.

docker run -it --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --device=/dev/kfd --device=/dev/dri --group-add video --ipc=host --shm-size 8G -p 7860:7860 -v ./data:/data samvv/sdwui:1.10.0-rocm

Most of the flags were taken from this page⁠.

FAQ

Will NVIDIA be supported?

If I get my hands on a NVIDIA GPU it will.

How do I get this working on my AMD Radeon 7800 XT?

Try to set the following environment variable. You might want to make this system-wide by appending it to /etc/environment for other apps and pass it directly to Docker with -e:

HSA_OVERRIDE_GFX_VERSION=11.0.0