Hi. I spent a long time trying to compile tensorflow-rocm but failed. However, you can also run the official rocm/tensorflow docker image, which works for me on Arch with no dependencies other than docker.
Link to the rocm tensorflow docker hub page
In summary:
sudo systemctl start systemd-timesyncd
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 $HOME/dockerx:/dockerx rocm/tensorflow:latest
My more in depth tutorial here: https://github.com/mpeschel10/test-tensorflow-rocm
Pinned Comments
lubosz commented on 2026-01-16 23:16 (UTC)
If you are interested in submitting patches, please use my github repo: https://github.com/lubosz/tensorflow-rocm