
AMD announced today that PyTorch machine learning development is now supported with the ROCm 5.7 series with select RDNA3 GPUs. The only mentioned RDNA3 GPUs are the Radeon RX 7900 XTX and the Radeon PRO W7900.
With the ROCm 5.7 series running atop Ubuntu Linux, AMD is now supporting ROCm with PyTorch for the Radeon RX 7900 XTX and PRO W7900. It's not clear if any other models like the Radeon RX 7900 XT or lower-end Radeon RX 700 / W7000 series hardware is supported yet and the ROCm release notes still do not indicate the RX 7900 XTX / W7900. It's well known that AMD is working to bring ROCm to more RDNA 3 GPUs so if the other models aren't working yet, hopefully they'll have it cleared up soon.
AMD's brief announcement of ROCm with PyTorch working for the W7900 and RX 7900 XTX can be found via the AMD Community Blog .