ROCm/docs/release/docker_support_matrix.md

Frameworks Support Matrix

The software support matrices for ROCm container releases is listed.

ROCm 5.6

PyTorch

Ubuntu+ rocm5.6_internal_testing +169530b

• ROCm5.6
• Python 3.8
• Torch 2.0.0
• Apex 0.1
• Torchvision 0.15.1
• Tensorboard 2.12.0
• MAGMA
• UCX 1.10.0
• OMPI 4.0.3
• OFED 5.4.3

CentOS7+ rocm5.6_internal_testing +169530b

• ROCm5.6
• Python 3.8
• Torch 2.0.0
• Apex 0.1
• Torchvision 0.15.1
• Tensorboard 2.12.0
• MAGMA

1.13 +bfeb431

• ROCm5.6
• Python 3.8
• Torch 1.13.1
• Apex 0.1
• Torchvision 0.14.0
• Tensorboard 2.12.0
• MAGMA
• UCX 1.10.0
• OMPI 4.0.3
• OFED 5.4.3

1.12 +05d5d04

• ROCm5.6
• Python 3.8
• Torch 1.12.1
• Apex 0.1
• Torchvision 0.13.1
• Tensorboard 2.12.0
• MAGMA
• UCX 1.10.0
• OMPI 4.0.3
• OFED 5.4.3

TensorFlow

tensorflow_develop-upstream-QA-rocm56 +c88a9f4

• ROCm5.6
• Python 3.9
• tensorflow-rocm 2.13.0
• OFED 5.3
• OMPI 4.0.7
• Horovod 0.27.0
• Tensorboard 2.12.0

r2.11-rocm-enhanced +5be4141

• ROCm5.6
• Python 3.9
• tensorflow-rocm 2.11.0
• OFED 5.3
• OMPI 4.0.7
• Horovod 0.27.0
• Tensorboard 2.11.2

r2.10-rocm-enhanced +72789a3

• ROCm5.6
• Python 3.9
• tensorflow-rocm 2.10.1
• OFED 5.3
• OMPI 4.0.7
• Horovod 0.27.0
• Tensorboard 2.10.1