Docker image support matrix#

Applies to Linux

2024-07-22

3 min read time

AMD validates and publishes ready-made PyTorch and TensorFlow images with ROCm backends on Docker Hub. The following Docker image tags and associated inventories are validated for ROCm 6.1.

PyTorch

Ubuntu 22.04

Tag

rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.1.2

Inventory

ROCm 6.1
Python 3.10
PyTorch 2.1.2
Apex 1.1.0
torchvision 0.16.1
TensorBoard 2.13.0
MAGMA
UCX 1.14.1
OMPI 4.1.5
OFED 5.4.3

Ubuntu 20.04

Tag

rocm/pytorch:rocm6.1_ubuntu20.04_py3.9_pytorch_2.1.2

Inventory

ROCm 6.1
Python 3.9
PyTorch 2.1.2
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

rocm/pytorch:rocm6.1_ubuntu20.04_py3.9_pytorch_2.0.1

Inventory

ROCm 6.1
Python 3.9
PyTorch 2.0.1
Apex 1.0.0
torchvision 0.15.2
TensorBoard 2.16.2
MAGMA
UCX 1.10.0
OMPI 4.0.3
OFED 5.4.3

Tag

rocm/pytorch:rocm6.1_ubuntu20.04_py3.9_pytorch_1.13.1

Inventory

ROCm 6.1
Python 3.9
PyTorch 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

rocm6.1_centos7_py3.9_pytorch_staging

Inventory

ROCm 6.1
Python 3.9
PyTorch 2.3.0
Apex 0.1
torchvision 0.18.0
MAGMA

TensorFlow

Ubuntu 20.04

Tag

rocm/tensorflow:rocm6.1-py3.10-tf2.15-dev

Inventory

ROCm 6.1
Python 3.9
tensorflow-rocm 2.15.0
TensorBoard 2.15.2

Tag

rocm/tensorflow:rocm6.1-py3.9-tf2.15-dev

Inventory

ROCm 6.1
Python 3.9
tensorflow-rocm 2.15.0
TensorBoard 2.15.2

Tag

rocm/tensorflow:rocm6.1-py3.9-tf2.14-dev

Inventory

ROCm 6.1
Python 3.9
tensorflow-rocm 2.14.0
TensorBoard 2.14.0

Tag

rocm/tensorflow:rocm6.1-py3.9-tf2.13-dev

Inventory

ROCm 6.1
Python 3.9
tensorflow-rocm 2.13.0
TensorBoard 2.13.0