Skip to content

721 Framework Docker image update [Develop] #699

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
prbasyal-amd merged 8 commits into from
Mar 27, 2026
Merged

721 Framework Docker image update [Develop] #699

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
5ff17e6
PyTorch Docker updated
prbasyal-amd Mar 27, 2026
04c1549
JAX Docker update
prbasyal-amd Mar 27, 2026
5182d98
Tensorflow update
prbasyal-amd Mar 27, 2026
dc774dc
Minor output change
prbasyal-amd Mar 27, 2026
04b27c6
Merge branch 'develop' into 721-docker-image
prbasyal-amd Mar 27, 2026
fbaebd3
Update pytorch-install.rst
prbasyal-amd Mar 27, 2026
8782ae2
Review feedback added
prbasyal-amd Mar 27, 2026
a4c1271
Merge branch '721-docker-image' of github.com:prbasyal-amd/rocm-insta…
prbasyal-amd Mar 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions docs/install/3rd-party/jax-install.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -90,12 +90,12 @@ For ``jax-community`` images, see `rocm/jax-community <https://hub.docker.com/r/

.. code-block:: shell

docker pull rocm/jax:rocm7.2-jax0.8.2-py3.12
docker pull rocm/jax:rocm7.2.1-jax0.8.2-py3.12

See
``rocm/jax:rocm7.2-jax0.8.2-py3.12``
``rocm/jax:rocm7.2.1-jax0.8.2-py3.12``
on `Docker Hub
<https://hub.docker.com/layers/rocm/jax/rocm7.2-jax0.8.2-py3.12/images/sha256-775462fde09a7a8f5685174bbb6353d8505f59b156df3d565edf98f890040526>`__.
<https://hub.docker.com/layers/rocm/jax/rocm7.2.1-jax0.8.2-py3.12/images/sha256-db79eadd49e9c02588b33f397b52c315a334de4b8e5896006b5dbc18ffd3f1fa>`__.

.. tab-item:: Python 3.11

Expand All @@ -106,9 +106,9 @@ For ``jax-community`` images, see `rocm/jax-community <https://hub.docker.com/r/
docker pull rocm/jax:rocm7.2-jax0.8.2-py3.11

See
``rocm/jax:rocm7.2-jax0.8.2-py3.11``
``rocm/jax:rocm7.2.1-jax0.8.2-py3.11``
on `Docker Hub
<https://hub.docker.com/layers/rocm/jax/rocm7.2-jax0.8.2-py3.11/images/sha256-fd11ee894df308770794ea389e8080d52ea4eff817aebc91cd49cf725bc36af8>`__.
<https://hub.docker.com/layers/rocm/jax/rocm7.2.1-jax0.8.2-py3.11/images/sha256-5ed631c65c49f0dda84d576fe18fa1d3e446497ad2ff3c58ca5369c29b7c2875>`__.

.. _build-jax-rocm-docker-image:

Expand Down Expand Up @@ -197,7 +197,7 @@ Follow these steps if you prefer to install ROCm manually on your host system or
.. code-block:: bash

+------------------------------------------------------------------------------+
| AMD-SMI 26.2.1+fc0010cf6a amdgpu version: 6.14.14 ROCm version: 7.2.0 |
| AMD-SMI 26.2.1+fc0010cf6a amdgpu version: 6.14.14 ROCm version: 7.2.1 |
| VBIOS version: 023.040.001.008.000001 |
| Platform: Linux Baremetal |
|-------------------------------------+----------------------------------------|
Expand Down
174 changes: 25 additions & 149 deletions docs/install/3rd-party/pytorch-install.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,134 +79,10 @@ Docker image support

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

.. tab-set::

.. tab-item:: PyTorch 2.10.0

.. tab-set::

.. tab-item:: Python 3.13

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu24.04_py3.13_pytorch_release_2.10.0

.. rubric:: Additional software components

.. list-table::
:header-rows: 1

* - Ubuntu
- Apex
- torchvision
- UCX
- Open MPI

* - 24.04
- `1.10.0+rocm7.2.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `0.25.0 <https://github.com/pytorch/vision/tree/v0.25.0>`__
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu24.04_py3.13_pytorch_release_2.10.0``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu24.04_py3.13_pytorch_release_2.10.0/images/sha256-1ff3c4572abffc475a0a91fa7a2c927af567d3cb78fba0e374c874ba26526c3a>`__.

.. tab-item:: Python 3.12

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.10.0

.. rubric:: Additional software components

.. list-table::
:header-rows: 1

* - Ubuntu
- Apex
- torchvision
- UCX
- Open MPI

* - 24.04
- `1.10.0+rocm7.2.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `0.25.0 <https://github.com/pytorch/vision/tree/v0.25.0>`__
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.10.0``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.10.0/images/sha256-bcb2efa1d43f66cc59fa45a1975d621a50b2b06d21a2abdfa2ba821ab32763e4>`__.

.. tab-item:: Python 3.11

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu22.04_py3.11_pytorch_release_2.10.0

.. rubric:: Additional software components

.. list-table::
:header-rows: 1

* - Ubuntu
- Apex
- torchvision
- UCX
- Open MPI

* - 22.04
- `1.10.0+rocm7.2.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `0.25.0 <https://github.com/pytorch/vision/tree/v0.25.0>`__
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu22.04_py3.11_pytorch_release_2.10.0``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu22.04_py3.11_pytorch_release_2.10.0/images/sha256-8ea1aa153bbd57b0b37ee20ed97dac5ee22edd0dfbd31f8937767e9c04f9e1d5>`__.

.. tab-item:: Python 3.10

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.10.0

.. rubric:: Additional software components

.. list-table::
:header-rows: 1

* - Ubuntu
- Apex
- torchvision
- UCX
- Open MPI

* - 22.04
- `1.10.0+rocm7.2.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `0.25.0 <https://github.com/pytorch/vision/tree/v0.25.0>`__
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.10.0``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.10.0/images/sha256-749fef598d1fe6e5771f38ba4768e850e3d35a2106df77e8e2982cb7c3e1312c>`__.

.. tab-item:: PyTorch 2.9.1

.. tab-set::
Expand All @@ -217,7 +93,7 @@ validated for ROCm 7.2.0.

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.9.1
docker pull rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.9.1

.. rubric:: Additional software components

Expand All @@ -231,23 +107,23 @@ validated for ROCm 7.2.0.
- Open MPI

* - 24.04
- `1.9.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `1.9.0+rocm7.2.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/>`__
- `0.24.0 <https://github.com/pytorch/vision/tree/v0.24.0>`__
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.9.1``
``rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.9.1``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.9.1/images/sha256-b18c14f18d85d1c3f3767a9ef8b0ff9a4e7a285c88703874933785a1d0e13383>`__.
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.9.1/images/sha256-b84bec4e2ae3c78e18a2bcf74fedc01e1c5d944721acf29cc08e09dac8356b3b>`__.

.. tab-item:: Python 3.10

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.9.1
docker pull rocm/pytorch:rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.9.1

.. rubric:: Additional software components

Expand All @@ -261,15 +137,15 @@ validated for ROCm 7.2.0.
- Open MPI

* - 22.04
- `1.9.0+rocm7.2 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `1.9.0+rocm7.2.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/>`__
- `0.24.0 <https://github.com/pytorch/vision/tree/v0.24.0>`__
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.9.1``
``rocm/pytorch:rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.9.1``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.9.1/images/sha256-376bfab5f4f680c8b4b843c6d0c5d1f0a04e5a84ec3e86728db8d11d79a9d1e3>`__.
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.9.1/images/sha256-0a347c18e55cc85e07e36b0c70cb8a9d23df2e8b8f69c3a38cff24935bddc460>`__.

.. tab-item:: PyTorch 2.8.0

Expand All @@ -281,7 +157,7 @@ validated for ROCm 7.2.0.

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.8.0
docker pull rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.8.0

.. rubric:: Additional software components

Expand All @@ -295,23 +171,23 @@ validated for ROCm 7.2.0.
- Open MPI

* - 24.04
- `1.8.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `1.8.0+rocm7.2.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/>`__
- `0.23.0 <https://github.com/pytorch/vision/tree/v0.23.0>`__
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.8.0``
``rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.8.0``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.8.0/images/sha256-fb6bfbeb27064cc70720b503192b3205ce601282b20c6258328ca5e1602bc3f0>`__.
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.8.0/images/sha256-e4e7900aadc927cd438329094183b05e6a7354a52d011219dd1573551a935d4c>`__.

.. tab-item:: Python 3.10

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.8.0
docker pull rocm/pytorch:rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.8.0

.. rubric:: Additional software components

Expand All @@ -325,15 +201,15 @@ validated for ROCm 7.2.0.
- Open MPI

* - 22.04
- `1.8.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `1.8.0+rocm7.2.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/>`__
- `0.23.0 <https://github.com/pytorch/vision/tree/v0.23.0>`__
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.8.0``
``rocm/pytorch:rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.8.0``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.8.0/images/sha256-c22b4ebff93e619d0c214ac1b1af037a358e8e667dfc955c92c245eaff5b9034>`__.
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.8.0/images/sha256-13f480a22b02151afbe4bacc0d5de93b4d210dbce5e604b422e6007e421febef>`__.

.. tab-item:: PyTorch 2.7.1

Expand All @@ -345,7 +221,7 @@ validated for ROCm 7.2.0.

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.7.1
docker pull rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.7.1

.. rubric:: Additional software components

Expand All @@ -359,23 +235,23 @@ validated for ROCm 7.2.0.
- Open MPI

* - 24.04
- `1.7.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `1.7.0+rocm7.2.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/>`__
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.7.1``
``rocm/pytorch:rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.7.1``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu24.04_py3.12_pytorch_release_2.7.1/images/sha256-ba4b669ebe67a6580dd5d0197e1927476fd26718850a2a9e85cd47829813b72e>`__.
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2.1_ubuntu24.04_py3.12_pytorch_release_2.7.1/images/sha256-d9ac66c8f1741b3f46460fe5e166bfc03cd7099da2eb74c261bbfc2552b600bc>`__.

.. tab-item:: Python 3.10

.. rubric:: Docker pull tag

.. code-block:: shell

docker pull rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.7.1
docker pull rocm/pytorch:rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.7.1

.. rubric:: Additional software components

Expand All @@ -389,15 +265,15 @@ validated for ROCm 7.2.0.
- Open MPI

* - 22.04
- `1.7.0 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2/>`__
- `1.7.0+rocm7.2.1 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/>`__
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__

See
``rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.7.1``
``rocm/pytorch:rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.7.1``
on `Docker Hub
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.7.1/images/sha256-d55f414155ff6bd4ea6f1c5803996689790ff32bd39eb0f8d2244ca91f267c9c>`__.
<https://hub.docker.com/layers/rocm/pytorch/rocm7.2.1_ubuntu22.04_py3.10_pytorch_release_2.7.1/images/sha256-27af38c5dc0cd92f08feaf3d35db3650d62166ad24b1022335fac256607fae01>`__.

.. _install_pytorch_wheels:
.. _using-wheels-package:
Expand Down
Loading
Loading