Skip to content

Commit 649d9c6

Browse files
yugang-amdyugang-amd
authored
7.1.0 Docker image support for Frameworks (#616)
* [7.1.0 Doc] Docker image support for Frameworks * fix links * fix docker links for JAX
1 parent 934f831 commit 649d9c6

File tree

3 files changed

+78
-73
lines changed

3 files changed

+78
-73
lines changed

‎docs/install/3rd-party/jax-install.rst‎

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ For ``jax-community`` images, see `rocm/jax-community <https://hub.docker.com/r/
7979

8080
.. tab-set::
8181

82-
.. tab-item:: JAX 0.6.0
82+
.. tab-item:: JAX 0.7.1
8383
:selected:
8484

8585
.. tab-set::
@@ -90,25 +90,25 @@ For ``jax-community`` images, see `rocm/jax-community <https://hub.docker.com/r/
9090

9191
.. code-block:: shell
9292
93-
docker pull rocm/jax:rocm7.0-jax0.6.0-py3.12
93+
docker pull rocm/jax:rocm7.1-jax0.7.1-py3.12
9494
9595
See
96-
``rocm/jax:rocm7.0-jax0.6.0-py3.12``
96+
``rocm/jax:rocm7.1-jax0.7.1-py3.12``
9797
on `Docker Hub
98-
<https://hub.docker.com/layers/rocm/jax/rocm7.0-jax0.6.0-py3.12/images/sha256-fc5445492b573b29d2b1d166e9dde0508bae34f2763ff3b045e9375ac9ce44eb>`__.
98+
<https://hub.docker.com/layers/rocm/jax/rocm7.1-jax0.7.1-py3.12/images/sha256-bc661fee582c591c4a9355e249afb912e8fd7dfe39f1fc02008821b1dc739bdf>`__.
9999

100-
.. tab-item:: Python 3.10
100+
.. tab-item:: Python 3.11
101101

102102
.. rubric:: Docker pull tag
103103

104104
.. code-block:: shell
105105
106-
docker pull rocm/jax:rocm7.0-jax0.6.0-py3.10
106+
docker pull rocm/jax:rocm7.1-jax0.7.1-py3.11
107107
108108
See
109-
``rocm/jax:rocm7.0-jax0.6.0-py3.10``
109+
``rocm/jax:rocm7.1-jax0.7.1-py3.11``
110110
on `Docker Hub
111-
<https://hub.docker.com/layers/rocm/jax/rocm7.0-jax0.6.0-py3.10/images/sha256-a75d73f696926c42434fa8910037e104b2d46211cf72a7f66f052a75e61f18f5>`__.
111+
<https://hub.docker.com/layers/rocm/jax/rocm7.1-jax0.7.1-py3.11/images/sha256-7e923ebfc117b2028c63991bfa1fbe274150ea94513c9530ece783f1a2f8a65e>`__.
112112

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

@@ -284,3 +284,4 @@ Expected output:
284284
.. code-block::
285285
286286
[0 1 2 3 4]
287+

‎docs/install/3rd-party/pytorch-install.rst‎

Lines changed: 35 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -82,11 +82,11 @@ Docker image support
8282

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

8787
.. tab-set::
8888

89-
.. tab-item:: PyTorch 2.7.1
89+
.. tab-item:: PyTorch 2.8.0
9090

9191
.. tab-set::
9292

@@ -96,7 +96,7 @@ validated for ROCm 7.0.0.
9696

9797
.. code-block:: shell
9898
99-
docker pull rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1
99+
docker pull rocm/pytorch:rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.8.0
100100
101101
.. rubric:: Additional software components
102102

@@ -110,23 +110,23 @@ validated for ROCm 7.0.0.
110110
- Open MPI
111111

112112
* - 24.04
113-
- `1.7.0a0+rocm7.0.git20a62dc5 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/>`__
114-
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
113+
- `1.8.0a0+rocm7.1.0.git3f26640c <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1/>`__
114+
- `0.23.0 <https://github.com/pytorch/vision/tree/v0.23.0>`__
115115
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
116116
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__
117117

118118
See
119-
``rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1``
119+
``rocm/pytorch:rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.8.0``
120120
on `Docker Hub
121-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.7.1/images/sha256-e9e3cd3a8e8b1450d884a2786977cbd822784ea73772c04c91505c0b0abe3843>`__.
121+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.8.0/images/sha256-3e917342db23b080cc7aa274321b4a7f33eb321e71b9607d69c0cb4deaaa8820>`__.
122122

123123
.. tab-item:: Python 3.10
124124

125125
.. rubric:: Docker pull tag
126126

127127
.. code-block:: shell
128128
129-
docker pull rocm/pytorch:rocm7.0_ubuntu22.04_py3.10_pytorch_release_2.7.1
129+
docker pull rocm/pytorch:rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.8.0
130130
131131
.. rubric:: Additional software components
132132

@@ -140,17 +140,17 @@ validated for ROCm 7.0.0.
140140
- Open MPI
141141

142142
* - 22.04
143-
- `1.7.0a0+rocm7.0.git20a62dc5 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/>`__
144-
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
143+
- `1.8.0a0+rocm7.1.0.git3f26640c <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1/>`__
144+
- `0.23.0 <https://github.com/pytorch/vision/tree/v0.23.0>`__
145145
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
146146
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__
147147

148148
See
149-
``rocm/pytorch:rocm7.0_ubuntu22.04_py3.10_pytorch_release_2.7.1``
149+
``rocm/pytorch:rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.8.0``
150150
on `Docker Hub
151-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.0_ubuntu22.04_py3.10_pytorch_release_2.7.1/images/sha256-1206916a14e585c42a7749bfd53506b293587aa66895e9b0d8b2c844730d3b0c>`__.
151+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.8.0/images/sha256-db07f47bba86735cf8346197fde49949a883fe6c62d44a0edf3d00450dba2be5>`__.
152152

153-
.. tab-item:: PyTorch 2.6.0
153+
.. tab-item:: PyTorch 2.7.1
154154

155155
.. tab-set::
156156

@@ -160,7 +160,7 @@ validated for ROCm 7.0.0.
160160

161161
.. code-block:: shell
162162
163-
docker pull rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.6.0
163+
docker pull rocm/pytorch:rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.7.1
164164
165165
.. rubric:: Additional software components
166166

@@ -174,23 +174,23 @@ validated for ROCm 7.0.0.
174174
- Open MPI
175175

176176
* - 24.04
177-
- `1.6.0+rocm7.0.git6596002f <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/>`__
178-
- `0.21.0 <https://github.com/pytorch/vision/tree/v0.21.0>`__
177+
- `1.7.0a0+rocm7.1.0.git20a62dc5 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1/>`__
178+
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
179179
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
180180
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__
181181

182182
See
183-
``rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.6.0``
183+
``rocm/pytorch:rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.7.1``
184184
on `Docker Hub
185-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.6.0/images/sha256-4969a34c735bea2ea0ad1bc3cbea94f889b49fbdecd2d559d96373fbc5291fca>`__.
185+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.7.1/images/sha256-1c695795a20f52ad242de549d58108041ded32d8af19b0ff3a2018f6e457a224>`__.
186186

187187
.. tab-item:: Python 3.10
188188

189189
.. rubric:: Docker pull tag
190190

191191
.. code-block:: shell
192192
193-
docker pull rocm/pytorch:rocm7.0_ubuntu22.04_py3.10_pytorch_release_2.6.0
193+
docker pull rocm/pytorch:rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.7.1
194194
195195
.. rubric:: Additional software components
196196

@@ -204,17 +204,17 @@ validated for ROCm 7.0.0.
204204
- Open MPI
205205

206206
* - 22.04
207-
- `1.6.0+rocm7.0.git6596002f <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/>`__
208-
- `0.21.0 <https://github.com/pytorch/vision/tree/v0.21.0>`__
207+
- `1.7.0a0+rocm7.1.0.git20a62dc5 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1/>`__
208+
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
209209
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
210210
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__
211211

212212
See
213-
``rocm/pytorch:rocm7.0_ubuntu22.04_py3.10_pytorch_release_2.6.0``
213+
``rocm/pytorch:rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.7.1``
214214
on `Docker Hub
215-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.0_ubuntu22.04_py3.10_pytorch_release_2.6.0/images/sha256-4b971f3b5908f0fdecaa8ccc05907a17f0b4a4f8e77208c9b9aa4f4eb7713403>`__.
215+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.7.1/images/sha256-63046938303e2770b7a6de2a94ffa46f7c9b805a84707b222272f54b8d0be781>`__.
216216

217-
.. tab-item:: PyTorch 2.5.1
217+
.. tab-item:: PyTorch 2.6.0
218218

219219
.. tab-set::
220220

@@ -224,7 +224,7 @@ validated for ROCm 7.0.0.
224224

225225
.. code-block:: shell
226226
227-
docker pull rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.5.1
227+
docker pull rocm/pytorch:rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.6.0
228228
229229
.. rubric:: Additional software components
230230

@@ -238,23 +238,23 @@ validated for ROCm 7.0.0.
238238
- Open MPI
239239

240240
* - 24.04
241-
- `1.5.0+rocm7.0.git118d7139 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/>`__
242-
- `0.20.1 <https://github.com/pytorch/vision/tree/v0.20.1>`__
241+
- `1.6.0+rocm7.1.0.git6596002f <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1/>`__
242+
- `0.21.0 <https://github.com/pytorch/vision/tree/v0.21.0>`__
243243
- `1.16.0+ds-5ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
244244
- `4.1.6-7ubuntu2 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__
245245

246246
See
247-
``rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.5.1``
247+
``rocm/pytorch:rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.6.0``
248248
on `Docker Hub
249-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.5.1/images/sha256-50e15654f110dfde046bc64080d5aa67e2c2ab505a57382e979552053af9473b>`__.
249+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.6.0/images/sha256-40c2e5583ac79bfed54f44dd4e57f3a00d45199e4cba0850aace06361d5cea63>`__.
250250

251251
.. tab-item:: Python 3.10
252252

253253
.. rubric:: Docker pull tag
254254

255255
.. code-block:: shell
256256
257-
docker pull rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.5.1
257+
docker pull rocm/pytorch:rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.6.0
258258
259259
.. rubric:: Additional software components
260260

@@ -268,15 +268,15 @@ validated for ROCm 7.0.0.
268268
- Open MPI
269269

270270
* - 22.04
271-
- `1.5.0+rocm7.0.git118d7139 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.0/>`__
272-
- `0.20.1 <https://github.com/pytorch/vision/tree/v0.20.1>`__
271+
- `1.6.0+rocm7.1.0.git6596002f <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1/>`__
272+
- `0.21.0 <https://github.com/pytorch/vision/tree/v0.21.0>`__
273273
- `1.12.1~rc2-1 <http://archive.ubuntu.com/ubuntu/pool/universe/u/ucx/>`__
274274
- `4.1.2-2ubuntu1 <http://archive.ubuntu.com/ubuntu/pool/universe/o/openmpi/>`__
275275

276276
See
277-
``rocm/pytorch:rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.5.1``
277+
``rocm/pytorch:rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.6.0``
278278
on `Docker Hub
279-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.0_ubuntu24.04_py3.12_pytorch_release_2.5.1/images/sha256-50e15654f110dfde046bc64080d5aa67e2c2ab505a57382e979552053af9473b>`__.
279+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.6.0/images/sha256-846520af9b22ce5aee1ddc9be7df3bec5257bf783ab8d084d33a0dcd48a430bf>`__.
280280

281281
.. _install_pytorch_wheels:
282282
.. _using-wheels-package:
@@ -743,3 +743,4 @@ Troubleshooting
743743
* How do you profile PyTorch workloads?
744744

745745
Use the PyTorch Profiler as described in :ref:`mi300x-pytorch-profiler` to profile GPU kernels on ROCm.
746+

0 commit comments

Comments
 (0)