Docker image support matrix

Docker image support matrix#

Applies to Linux

2024-06-28

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 6.1.

Tag

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

Inventory
Tag

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

Inventory
Tag

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

Inventory
Tag

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

Inventory
Tag

rocm6.1_centos7_py3.9_pytorch_staging

Inventory
Tag

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

Inventory
Tag

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

Inventory
Tag

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

Inventory
Tag

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

Inventory