Docker image support matrix

Applies to Linux

2024-01-11

3 min read time

AMD validates and publishes PyTorch and TensorFlow containers on Docker Hub. The following tags, and associated inventories, are validated with ROCm 5.7.

PyTorch

Ubuntu 22.04

Tag: rocm/pytorch:rocm5.7_ubuntu22.04_py3.10_pytorch_2.0.1

• Inventory:

  • ROCm 5.7

  • Python 3.10

  • Torch 2.0.1

  • Apex 0.1

  • Torchvision 0.15.0

  • Tensorboard 2.14.0

  • MAGMA

  • UCX 1.10.0

  • OMPI 4.0.3

  • OFED 5.4.3

Ubuntu 20.04

Tag: rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_staging

• Inventory:

  • ROCm 5.7

  • Python 3.9

  • Torch 2.1.0

  • Apex 0.1

  • Torchvision 0.16.0

  • Tensorboard 2.14.0

  • MAGMA

  • UCX 1.10.0

  • OMPI 4.0.3

  • OFED 5.4.3

Tag: Ubuntu rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_1.12.1

• Inventory:

  • ROCm 5.7

  • Python 3.9

  • Torch 1.12.1

  • Apex 0.1

  • Torchvision 0.13.1

  • Tensorboard 2.14.0

  • MAGMA

  • UCX 1.10.0

  • OMPI 4.0.3

  • OFED 5.4.3

Tag: Ubuntu rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_1.13.1

• Inventory:

  • ROCm 5.7

  • Python 3.9

  • 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

Tag: Ubuntu rocm/pytorch:rocm5.7_ubuntu20.04_py3.9_pytorch_2.0.1

• Inventory:

  • ROCm 5.7

  • Python 3.9

  • Torch 2.0.1

  • Apex 0.1

  • Torchvision 0.15.2

  • Tensorboard 2.14.0

  • MAGMA

  • UCX 1.10.0

  • OMPI 4.0.3

  • OFED 5.4.3

CentOS 7

Tag: rocm/pytorch:rocm5.7_centos7_py3.9_pytorch_staging

• Inventory:

• ROCm 5.7 (yum)

• Python 3.9

• Torch 2.1.0

• Apex 0.1

• Torchvision 0.16.0

• MAGMA

TensorFlow

Ubuntu 20.04

Tag: rocm5.7-tf2.12-dev

• Inventory:

  • ROCm 5.7

  • Python 3.9

  • tensorflow-rocm 2.12.1

  • Tensorboard 2.12.3

Tag: rocm5.7-tf2.13-dev

• Inventory:

  • ROCm 5.7

  • Python 3.9

  • tensorflow-rocm 2.13.0

  • Tensorboard 2.13.0