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 Halo" APUs.
At Computex 2025 we committed to making ROCm a true cross-platform, developer first stack. Today we take an important step on that journey by enabling native PyTorch support on Windows and Linux for users with AMD Radeon 7000 & 9000 GPUs and Ryzen AI 300 & Ryzen AI Max APUs.
This preview release means developers can now run AI models directly on AMD hardware in the Windows environment. This preview is designed to be a foundation for developers to help shape what comes next as performance and feature coverage continue to improve and evolve.
via AMD
AMD ROCm 7 Software Stack | Image Credits: AMD
ROCm 6.4.4 is being released as a preview, which will allow developers to run AI models on the supported hardware in the Windows environment. The preview will pave the path for future upgrades, which will focus on adding further performance and features to ROCm on Windows. Over on the enterprise side, AMD has already released ROCm 7 which further accelerates AI and productivity innovation on Instinct / EPYC hardware.
| RX 9000 | RX 7000 | Radeon PRO | Ryzen AI 300 |
|---|---|---|---|
| AMD Radeon AI PRO R9700 | AMD Radeon RX 7900 XTX | AMD Radeon PRO W7900 | AMD Ryzen AI Max+ 395 |
| AMD Radeon RX 9070 XT | AMD Radeon RX 7900 XT | AMD Radeon PRO W7900 Dual Slot | AMD Ryzen AI Max 390 |
| AMD Radeon RX 9070 | AMD Radeon RX 7900 GRE | AMD Ryzen AI Max 385 | |
| AMD Radeon RX 9070 GRE | AMD Ryzen AI 9 HX 375 | ||
| AMD Radeon RX 9060 XT | AMD Ryzen AI 9 HX 370 | ||
| AMD Ryzen AI 9 365 |
For those who want to get started with ROCm, you can do so here. AMD also has full documentation for PyTorch on Windows/Linux here.
About the author: A Software Engineer by training and a PC enthusiast by passion, Hassan Mujtaba serves as Wccftech's Senior Editor for hardware section. With years of experience in the industry, he specializes in deep-dive technical analysis of next-generation CPU and GPU architectures, motherboards, and cooling solutions. His work involves not only breaking news on upcoming technologies but also extensive hands-on reviews and benchmarking.
Follow Wccftech on Google to get more of our news coverage in your feeds.
