Intro
This repo hosts the docker images with ROCm backend support for extra architectures.
Supported Architectures:
| Image | Description |
|---|---|
| gfx803 | Uses workaround: https://github.com/ROCmSoftwarePlatform/rocBLAS/issues/1218 |
| gfx900 | From upstream. |
| gfx906 | From upstream. |
| gfx908 | From upstream. |
| gfx90a | From upstream. |
| gfx1010 | Fixed https://github.com/ROCmSoftwarePlatform/Tensile/issues/1757 |
| gfx1011 | Use drun -e HSA_OVERRIDE_GFX_VERSION=10.1.0 ... |
| gfx1012 | Use drun -e HSA_OVERRIDE_GFX_VERSION=10.1.0 ... |
| gfx1030 | From upstream. |
| gfx1031 | Use drun -e HSA_OVERRIDE_GFX_VERSION=10.3.0 ... |
| gfx1032 | Use drun -e HSA_OVERRIDE_GFX_VERSION=10.3.0 ... |
| gfx1100 | From upstream. |
| gfx1101 | From upstream. |
| gfx1102 | From upstream. |
Usage
Recommended aliases
alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $(pwd):/current'
Images
See each image README for details.
| Image | Description | Docs |
|---|---|---|
ulyssesrr/rocm-xtra-dev |
Base image for development | README |
ulyssesrr/rocm-xtra-llamacpp |
llama.cpp + PR#1087 | README |
ulyssesrr/rocm-xtra-pytorch |
PyTorch (OpenBLAS/MAGMA) | README |
ulyssesrr/rocm-xtra-stable-diffusion-webui |
AUTOMATIC1111’s Stable Diffusion Web UI | README |
Builder Images
TODO