AMD?

[CapsAdmin]

I'm using Ubuntu 20 and a rx 6900 XT.

I tried setting --extra-index-url in Dockerfile to https://download.pytorch.org/whl/rocm5.1.1 which I believe is correct for AMD systems.

I then had to modify the run script to be

docker run --rm \
        --group-add $(getent group render | cut -d':' -f 3) \
        --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined \
        -v huggingface:/home/huggingface/.cache/huggingface \
        -v "$PWD"/output:/home/huggingface/output \
        "$CWD" "$@"`

Which was suggested by https://github.com/RadeonOpenCompute/ROCm-docker

Then it works.

I'm not sure if this is expected, but it takes just under 2 min to generate an image. I can hear gpu affect my speakers (the same way games running at a high fps can do) the last 20 sec which could be a hint at some gpu usage at the end. It happens when the progress bar iteration thing shows up.