Linux support matrices by ROCm version#

PyTorch on Windows#

Compatible OS, GPU, and framework support matrices for the latest ROCm release.

To rollback support matrices and install instructions for previous versions, click Version List located at the top-right corner of the screen, or select the version (v:) menu on the bottom-left.

OS support matrix#

OS

Supported

Windows 11

Yes

GPU support matrix#

ROCm Version

Supported Architectures

Supported AMD Radeon™ Hardware

6.4.4*

gfx1150
gfx1151

AMD Ryzen AI Max+ 395
AMD Ryzen AI Max 390
AMD Ryzen AI Max 385
AMD Ryzen AI 9 HX 375
AMD Ryzen AI 9 HX 370
AMD Ryzen AI 9 365

Framework + ROCm support matrices#

View the ROCm support matrices for PyTorch.

PyTorch + ROCm support matrix#

PyTorch Version

ROCm Version

Python Version

Comments

2.8

6.4.4*

3.12

Official production support. See Install PyTorch for Ryzen APUs.

AI Data Types

  • FP16

*Pytorch on Windows includes ROCm 6.4.4 components; however, the entire ROCm stack is not yet supported on Windows.