I think PyTorch support is decent, I use it on my Radeon VII a lot. - You can compile on https://lernapparat.de/pytorch-rocm/ https://lernapparat.de/pytorch-roc

I think PyTorch support is decent, I use it on my Radeon VII a lot. - You can compile on https://lernapparat.de/pytorch-rocm/ https://lernapparat.de/pytorch-rocm/ (disclaimer my own link) - Arch has support out of the box: https://aur.archlinux.org/packages/python-pytorch-rocm/ https://aur.archlinux.org/packages/python-pytorch-rocm/ - AMD also has a ROCm docker with PyTorch (https://rocmdocs.amd.com/en/latest/Deep_learning/Deep-learning.html#recommended-install-using-published-pytorch-rocm-docker-image https://rocmdocs.amd.com/en/latest/Deep_learning/Deep-learni...) - If you know the secret URL (which you can infer from the other URLs on the front page), there are also test wheels for rocm3.8. So AMD got a lot of heat for not supporting Navi (RDNA) with ROCm, but it seems that they are weeding out the things keeping people from running it ( https://github.com/ROCmSoftwarePlatform/pytorch/issues/718 https://github.com/ROCmSoftwarePlatform/pytorch/issues/718 and the links in that look like gfx10 is almost there for rocBlas and MIOpen). We'll see what ROCm 3.9 will bring and what the state of big navi is.