ulyssesrr/docker-rocm-xtra

ROCm docker images with fixes/support for extra architectures, such as gfx803/gfx1010.

Intro

This repo hosts the docker images with ROCm backend support for extra architectures.

Supported Architectures:

Image	Description
gfx803	Uses workaround: ROCm/rocBLAS#1218
gfx900	From upstream.
gfx906	From upstream.
gfx908	From upstream.
gfx90a	From upstream.
gfx1010	Fixed ROCm/Tensile#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

docker-rocm-xtra's Issues

May it support ROCm 5.6 and up?

torchaudio

Hi,
The only image that worked for me out of the box is yours, all i'm missing is torchaudio.
Could you please add torchaudio?
Thank you

Docker mounted folders and file permission issues

Hi, I mounted the image and it works, but the folders and files it creates when mounting are read only, or have some permission restrictions that prevent me from deleting them or moving files into the folders, creating, etc, or even it didn't allowed Ubuntu to create thumbnails.

The workaround I used was to run sudo nautilus in Ubuntu (currently I'm on Arch). Is there a way to run it without this issue?

Regards.