AMD has finally enabled PyTorch support on Windows for Radeon RX 9000, RX 7000 GPUs & Ryzen AI APUs with ROCm 6.4.4.

AMD Radeon RX 9000, RX 7000 GPUs, & Ryzen AI APUs Now Support PyTorch On Windows Thanks To ROCm 6.4.4

At Computex 2025, AMD promised to bring PyTorch support for Windows through ROCm to its consumer GPU/APU lineup. Today, AMD is finally fulfilling its promise with ROCm 6.4.4, which adds PyTorch support on Windows for a range of consumer-centric GPUs and APUs. Today's support isn't limited to just Windows, as Linux users will also receive the same level of support.

In terms of hardware, PyTorch on Windows/Linux will be enabled through ROCm 6.4.4 on the latest Radeon RX 9000 (RDNA 4), Radeon RX 7000 (RDNA 3) GPUs, along with Ryzen AI 300 "Strix", & Ryzen AI MAX "Strix Hal

See Full Page