Skip to content

Commit

Permalink
Docker image support matrix for 6.2 (#258)
Browse files Browse the repository at this point in the history
* update Docker image support matrix for 6.2

* update docker image support matrix for 6.2

rm unpublished dockers

fix v

* update tf docker image tags
  • Loading branch information
peterjunpark authored Aug 9, 2024
1 parent 50d4244 commit c33e179
Showing 1 changed file with 117 additions and 41 deletions.
158 changes: 117 additions & 41 deletions docs/reference/docker-image-support-matrix.rst
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ Docker image support matrix

AMD validates and publishes ready-made `PyTorch <https://hub.docker.com/r/rocm/pytorch>`_ and
`TensorFlow <https://hub.docker.com/r/rocm/tensorflow>`_ images with ROCm backends on Docker Hub.
The following Docker image tags and associated inventories are validated for ROCm 6.1.
The following Docker image tags and associated inventories are validated for ROCm 6.2.

.. tab-set::

Expand All @@ -21,34 +21,73 @@ The following Docker image tags and associated inventories are validated for ROC
.. tab-item:: Ubuntu 22.04

Tag
`rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.1.2 <https://hub.docker.com/layers/rocm/pytorch/rocm6.1_ubuntu22.04_py3.10_pytorch_2.1.2/images/sha256-f6ea7cee8aae299c7f6368187df7beed29928850c3929c81e6f24b34271d652b>`_
`rocm/pytorch:rocm6.2_ubuntu22.04_py3.10_pytorch_release_2.3.0 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu22.04_py3.10_pytorch_release_2.3.0/images/sha256-931d3e3dcebe6c6fab84adf16cfca3e1d1449100df7c881a46fccd06f6c9bc1c>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.10 <https://www.python.org/downloads/release/python-31013/>`_
* `PyTorch 2.1.2 <https://github.com/ROCm/pytorch/tree/release/2.1>`_
* `Apex 1.1.0 <https://github.com/ROCm/apex/tree/release/1.1.0>`_
* `torchvision 0.16.1 <https://github.com/pytorch/vision/tree/v0.16.1>`_
* `PyTorch 2.3.0 <https://github.com/ROCm/pytorch/tree/release/2.3>`_
* `Apex 1.3.0 <https://github.com/ROCm/apex/tree/release/1.3.0>`_
* `torchvision 0.18.0 <https://github.com/pytorch/vision/tree/v0.18.0>`_
* `TensorBoard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.14.1 <https://github.com/openucx/ucx/tree/v1.14.1>`_
* `OMPI 4.1.5 <https://github.com/open-mpi/ompi/tree/v4.1.5>`_
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_

.. raw:: html

<hr>


Tag
`rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_release_2.2.1 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu22.04_py3.9_pytorch_release_2.2.1/images/sha256-86b214d2e4b380c3a6fdc8e161d5cc0a154a567880a2377f636ebf80611dfff7>`_

Inventory
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `PyTorch 2.2.1 <https://github.com/ROCm/pytorch/tree/release/2.2.1>`_
* `Apex 1.2.0 <https://github.com/ROCm/apex/tree/release/1.2.0>`_
* `torchvision 0.17.1 <https://github.com/pytorch/vision/tree/v0.17.1>`_
* `TensorBoard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.14.1 <https://github.com/openucx/ucx/tree/v1.14.1>`_
* `OMPI 4.1.5 <https://github.com/open-mpi/ompi/tree/v4.1.5>`_
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_

.. raw:: html

<hr>

Tag
`rocm/pytorch:rocm6.2_ubuntu22.04_py3.9_pytorch_1.13.1 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu22.04_py3.9_pytorch_release_1.13.1/images/sha256-fe417e67704625038ed782bf48ac892a1451721d86eef5f8b28b5fe16d320e0d>`_

Inventory
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `PyTorch 1.13.1 <https://github.com/ROCm/pytorch/tree/release/1.13.1>`_
* `Apex 1.0.0 <https://github.com/ROCm/apex/tree/release/1.0.0>`_
* `torchvision 0.14.0 <https://github.com/pytorch/vision/tree/v0.14.0>`_
* `TensorBoard 2.17.0 <https://github.com/tensorflow/tensorboard/tree/2.17>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.14.1 <https://github.com/openucx/ucx/tree/v1.14.1>`_
* `OMPI 4.1.5 <https://github.com/open-mpi/ompi/tree/v4.1.5>`_
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_

.. tab-item:: Ubuntu 20.04

Tag
`rocm/pytorch:rocm6.1_ubuntu20.04_py3.9_pytorch_2.1.2 <https://hub.docker.com/layers/rocm/pytorch/rocm6.1_ubuntu20.04_py3.9_pytorch_2.1.2/images/sha256-96ba7b33bc7fa37b88c5ed550488960dd1dcedabaa8a9c17a4fd62e6c50e7574>`_
`rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.3.0 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.3.0/images/sha256-a1b2be0e705b02c25a3cf7fdaa991afea68deaebcafa58ef1872ce961713617c>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `PyTorch 2.1.2 <https://github.com/ROCm/pytorch/tree/release/2.1>`_
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
* `torchvision 0.16.0 <https://github.com/pytorch/vision/tree/release/0.16>`_
* `TensorBoard 2.14.0 <https://github.com/tensorflow/tensorboard/tree/2.14>`_
* `PyTorch 2.3.0 <https://github.com/ROCm/pytorch/tree/release/2.3>`_
* `Apex 1.3.0 <https://github.com/ROCm/apex/tree/release/1.3.0>`_
* `torchvision 0.18.0 <https://github.com/pytorch/vision/tree/v0.18.0>`_
* `TensorBoard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
* `UCX 1.10.1 <https://github.com/openucx/ucx/tree/v1.10.0>`_
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_

Expand All @@ -57,15 +96,15 @@ The following Docker image tags and associated inventories are validated for ROC
<hr>

Tag
`rocm/pytorch:rocm6.1_ubuntu20.04_py3.9_pytorch_2.0.1 <https://hub.docker.com/layers/rocm/pytorch/rocm6.1_ubuntu20.04_py3.9_pytorch_2.0.1/images/sha256-77656adb5e8254053ef05879bfb0e62076172d0b3e89d06e18cc6b7071fdfd03>`_
`rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_2.2.1 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.2.1/images/sha256-83ee72d83356c36df640133d088189f16c8f119ec9c6569a873dfaa8de161f01>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `PyTorch 2.0.1 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/release/2.0>`_
* `Apex 1.0.0 <https://github.com/ROCm/apex/tree/release/1.0.0>`_
* `torchvision 0.15.2 <https://github.com/pytorch/vision/tree/release/0.15>`_
* `TensorBoard 2.16.2 <https://github.com/tensorflow/tensorboard/tree/2.16.2>`_
* `PyTorch 2.2.1 <https://github.com/ROCm/pytorch/tree/release/2.2>`_
* `Apex 1.2.0 <https://github.com/ROCm/apex/tree/1.2.0>`_
* `torchvision 0.17.1 <https://github.com/pytorch/vision/tree/v0.17>`_
* `TensorBoard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
Expand All @@ -76,15 +115,15 @@ The following Docker image tags and associated inventories are validated for ROC
<hr>

Tag
`rocm/pytorch:rocm6.1_ubuntu20.04_py3.9_pytorch_1.13.1 <https://hub.docker.com/layers/rocm/pytorch/rocm6.1_ubuntu20.04_py3.9_pytorch_1.13.1/images/sha256-9a01c527a05c53fb9966159650a30b10c134c987825dce0646fbf36a24bfba7c>`_
`rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.1.2 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu20.04_py3.9_pytorch_release_2.1.2/images/sha256-58186da550e3d83c5b598ce0c1f581206eabd82c85bd77d22b34f5695d749762>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `PyTorch 2.0.1 <https://github.com/ROCmSoftwarePlatform/pytorch/tree/release/2.0>`_
* `Apex 0.1 <https://github.com/ROCmSoftwarePlatform/apex/tree/v0.1>`_
* `torchvision 0.15.2 <https://github.com/pytorch/vision/tree/release/0.15>`_
* `TensorBoard 2.14.0 <https://github.com/tensorflow/tensorboard/tree/2.14>`_
* `PyTorch 2.1.2 <https://github.com/ROCm/pytorch/tree/release/2.1>`_
* `Apex 1.1.0 <https://github.com/ROCm/apex/tree/1.1.0>`_
* `torchvision 0.16.1 <https://github.com/pytorch/vision/tree/release/0.16>`_
* `TensorBoard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
Expand All @@ -94,56 +133,93 @@ The following Docker image tags and associated inventories are validated for ROC

<hr>

Tag
`rocm/pytorch:rocm6.2_ubuntu20.04_py3.9_pytorch_1.13.1 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2_ubuntu20.04_py3.9_pytorch_release_1.13.1/images/sha256-9338a4fb9cac31fa25f1c7c8907e6ab1bbf8b57f4e04b7af2ae4dbf7c7d36dae>`_

Inventory
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `PyTorch 1.13.1 <https://github.com/ROCm/pytorch/tree/release/1.13.1>`_
* `Apex 1.0.0 <https://github.com/ROCm/apex/tree/1.0.0>`_
* `torchvision 0.14.0 <https://github.com/pytorch/vision/tree/release/0.14.0>`_
* `TensorBoard 2.17.0 <https://github.com/tensorflow/tensorboard/tree/2.17.0>`_
* `MAGMA <https://bitbucket.org/icl/magma/src/master/>`_
* `UCX 1.10.0 <https://github.com/openucx/ucx/tree/v1.10.0>`_
* `OMPI 4.0.3 <https://github.com/open-mpi/ompi/tree/v4.0.3>`_
* `OFED 5.4.3 <https://content.mellanox.com/ofed/MLNX_OFED-5.3-1.0.5.0/MLNX_OFED_LINUX-5.3-1.0.5.0-ubuntu20.04-x86_64.tgz>`_

.. tab-item:: TensorFlow

.. tab-set::

.. tab-item:: Ubuntu 20.04

Tag
`rocm/tensorflow:rocm6.1-py3.10-tf2.15-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.1-py3.10-tf2.15-dev/images/sha256-5bb6212c86376d3333be0fc170fef785d8a13a2dc2a3e33649c9a4d86ec38d70>`_
`rocm/tensorflow:rocm6.2-py3.9-tf2.16-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.2-py3.9-tf2.16-dev/images/sha256-dcdeafe0dcb5b5160c7ab7ef860dc29a95f2d2dd691946497ab6fb549cde8497>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-31013/>`_
* `tensorflow-rocm 2.15.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/tensorflow_rocm-2.15.0-cp310-cp310-manylinux2014_x86_64.whl>`_
* `TensorBoard 2.15.2 <https://github.com/tensorflow/tensorboard/tree/2.15.2>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `tensorflow-rocm 2.16.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/>`_
* `TensorBoard 2.16.2 <https://github.com/tensorflow/tensorboard/tree/2.16.2>`_

.. raw:: html

<hr>

Tag
`rocm/tensorflow:rocm6.1-py3.9-tf2.15-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.1-py3.9-tf2.15-dev/images/sha256-1435fe5c762ae1a331b9c210c48553787c05f12145d7eeb3e140b98467d04911>`_
`rocm/tensorflow:rocm6.2-py3.9-tf2.16-runtime <https://hub.docker.com/layers/rocm/tensorflow/rocm6.2-py3.9-tf2.16-runtime/images/sha256-ea1f05be5f618111ad0edbf25458fc96e02bc596859cf8c7ddbbf7c797fa22b3>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `tensorflow-rocm 2.15.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/tensorflow_rocm-2.15.0-cp310-cp310-manylinux2014_x86_64.whl>`_
* `tensorflow-rocm 2.16.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/>`_
* `TensorBoard 2.16.2 <https://github.com/tensorflow/tensorboard/tree/2.16.2>`_

.. raw:: html

<hr>

Tag
`rocm/tensorflow:rocm6.2-py3.9-tf2.15-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.2-py3.9-tf2.15-dev/images/sha256-1a28f5735a719e2a6ef076523ce76fa308663ad12f0b5530666468a2b775666f>`_

Inventory
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `tensorflow-rocm 2.15.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/>`_
* `TensorBoard 2.15.2 <https://github.com/tensorflow/tensorboard/tree/2.15.2>`_

.. raw:: html

<hr>

Tag
`rocm/tensorflow:rocm6.1-py3.9-tf2.14-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.1-py3.9-tf2.14-dev/images/sha256-e871c1cbe9150787395d103c05bf8c923c8457ded13dbfc2f479ecab9c4661d8>`_
`rocm/tensorflow:rocm6.2-py3.9-tf2.15-runtime <https://hub.docker.com/layers/rocm/tensorflow/rocm6.2-py3.9-tf2.15-runtime/images/sha256-81ef38ce067666ab2a4ba3cff8f5803f8596b4d7395169b4f0e2946ba2a403f6>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `tensorflow-rocm 2.14.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/tensorflow_rocm-2.14.0-cp39-cp39-manylinux2014_x86_64.whl>`_
* `TensorBoard 2.14.0 <https://github.com/tensorflow/tensorboard/tree/2.14.0>`_
* `tensorflow-rocm 2.15.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/>`_
* `TensorBoard 2.15.2 <https://github.com/tensorflow/tensorboard/tree/2.15.2>`_

.. raw:: html

<hr>

Tag
`rocm/tensorflow:rocm6.1-py3.9-tf2.13-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.1-py3.9-tf2.13-dev/images/sha256-f221f0838f5c2320052bd54de640494e578fb860afac07144e91c8a730304e8f>`_
`rocm/tensorflow:rocm6.2-py3.9-tf2.14-dev <https://hub.docker.com/layers/rocm/tensorflow/rocm6.2-py3.9-tf2.14-dev/images/sha256-ad1fbaed5b9f2085d2716468147aca8d390e0e99470960e7b8f7e11d0286e80f>`_

Inventory
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `tensorflow-rocm 2.14.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/>`_
* `TensorBoard 2.14.1 <https://github.com/tensorflow/tensorboard/tree/2.15.2>`_

Tag
`rocm/tensorflow:rocm6.2-py3.9-tf2.14-runtime <https://hub.docker.com/layers/rocm/tensorflow/rocm6.2-py3.9-tf2.14-runtime/images/sha256-ed0ca0548ba140253e23ef683440e144e90e309e26d208ad2a84b5d6d5ddd95a>`_

Inventory
* `ROCm 6.1 <https://repo.radeon.com/rocm/apt/6.1/>`_
* `ROCm 6.2 <https://repo.radeon.com/rocm/apt/6.2/>`_
* `Python 3.9 <https://www.python.org/downloads/release/python-3918/>`_
* `tensorflow-rocm 2.13.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/tensorflow_rocm-2.13.1-cp39-cp39-manylinux2014_x86_64.whl>`_
* `TensorBoard 2.13.0 <https://github.com/tensorflow/tensorboard/tree/2.13.0>`_
* `tensorflow-rocm 2.14.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-6.2/>`_
* `TensorBoard 2.14.1 <https://github.com/tensorflow/tensorboard/tree/2.15.2>`_

0 comments on commit c33e179

Please sign in to comment.