AMD Brings Native PyTorch Support to Windows with ROCm 6.4.4
AMD has officially launched the ROCm 6.4.4 software update to bring native support for PyTorch to Windows on its consumer-grade Radeon GPUs and Ryzen AI APUs. This completes AMD's promise made during Computex 2022 to take its ROCm software stack to a cross-platform level for developer use. The same support is now extended to Linux users.
AMD Hardware Supported
PyTorch support via ROCm 6.4.4 shall be applicable to the following AMD hardware on any of Windows or Linux:
- Radeon RX 9000 Series GPUs (RDNA 4)
- Radeon RX 7000 Series GPUs (RDNA 3)
- Ryzen AI 300 "Strix" APUs
- Ryzen AI MAX "Strix Halo" APUs
Foundational Preview for Developers Paid ROCm 6.4.4 Release
This ROCm 6.4.4 release is under the Preview designation. Its major aim is to provide a foundation for developers to run AI models directly on AMD hardware within the Windows environment. AMD opines that this initial release, made available, is expected to support the development toward an evolving AMD software stack for the sake of performance enhancement and coverage of features.
AMD said that "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."
Enterprise Vs. Consumer Stacks
This update for consumer hardware differs from AMD's enterprise-oriented route. The recent ROCm 7 that AMD has launched is meant to heighten AI innovations and productivity on their high-performance Instinct and EPYC hardware for data centers and professional workstation market segmentation.
