Ctrl+K

Docker image support matrix

Docker image support matrix#

Applies to Linux

2024-08-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.2.

Tag

rocm/pytorch:rocm6.2_ubuntu22.04_py3.10_pytorch_release_2.3.0

Inventory
Tag

rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_release_2.2.1

Inventory
Tag

rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_1.13.1

Inventory
Tag

rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.3.0

Inventory
Tag

rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_2.2.1

Inventory
Tag

rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.1.2

Inventory
Tag

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

Inventory
Tag

rocm/tensorflow:rocm6.2-py3.9-tf2.16-dev

Inventory
Tag

rocm/tensorflow:rocm6.2-py3.9-tf2.16-runtime

Inventory
Tag

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

Inventory
Tag

rocm/tensorflow:rocm6.2-py3.9-tf2.15-runtime

Inventory
Tag

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

Inventory
Tag

rocm/tensorflow:rocm6.2-py3.9-tf2.14-runtime

Inventory