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