Skip to content

Commit 5228e32

Browse files
prbasyal-amdprbasyal-amd
committed
PyTorch docker update
1 parent ba6f66f commit 5228e32

File tree

1 file changed

+34
-34
lines changed

1 file changed

+34
-34
lines changed

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

Lines changed: 34 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.1.0.
85+
validated for ROCm 7.1.1.
8686

8787
.. tab-set::
8888

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

9191
.. tab-set::
9292

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

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

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

112112
* - 24.04
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>`__
113+
- `1.9.0+rocm7.1.1.git07c3ee53 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/>`__
114+
- `0.24.0 <https://github.com/pytorch/vision/tree/v0.24.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.1_ubuntu24.04_py3.12_pytorch_release_2.8.0``
119+
``rocm/pytorch:rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.9.1``
120120
on `Docker Hub
121-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.8.0/images/sha256-3e917342db23b080cc7aa274321b4a7f33eb321e71b9607d69c0cb4deaaa8820>`__.
121+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.9.1/images/sha256-683765a52c61341e1674fe730ab3be861a444a45a36c0a8caae7653a08a0e208>`__.
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.1_ubuntu22.04_py3.10_pytorch_release_2.8.0
129+
docker pull rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.9.1
130130
131131
.. rubric:: Additional software components
132132

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

142142
* - 22.04
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>`__
143+
- `1.9.0+rocm7.1.1.git07c3ee53 <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/>`__
144+
- `0.24.0 <https://github.com/pytorch/vision/tree/v0.24.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.1_ubuntu22.04_py3.10_pytorch_release_2.8.0``
149+
``rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.9.1``
150150
on `Docker Hub
151-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.8.0/images/sha256-db07f47bba86735cf8346197fde49949a883fe6c62d44a0edf3d00450dba2be5>`__.
151+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.9.1/images/sha256-079ef6d4ef803ffb4191983f67149a70dddb0b4e58fb59fbf304df2829ed173e>`__.
152152

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

155155
.. tab-set::
156156

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

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

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

176176
* - 24.04
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>`__
177+
- `1.8.0a0+rocm7.1.1.git3f26640c <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/>`__
178+
- `0.23.0 <https://github.com/pytorch/vision/tree/v0.23.0>`__
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.1_ubuntu24.04_py3.12_pytorch_release_2.7.1``
183+
``rocm/pytorch:rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.8.0``
184184
on `Docker Hub
185-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.7.1/images/sha256-1c695795a20f52ad242de549d58108041ded32d8af19b0ff3a2018f6e457a224>`__.
185+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.8.0/images/sha256-a0bc4dc0e614bb80635da8af3ee3ae37434eb09d33dae6e668f769188abc8c2f>`__.
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.1_ubuntu22.04_py3.10_pytorch_release_2.7.1
193+
docker pull rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.8.0
194194
195195
.. rubric:: Additional software components
196196

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

206206
* - 22.04
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>`__
207+
- `1.8.0a0+rocm7.1.1.git3f26640c <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/>`__
208+
- `0.23.0 <https://github.com/pytorch/vision/tree/v0.23.0>`__
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.1_ubuntu22.04_py3.10_pytorch_release_2.7.1``
213+
``rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.8.0``
214214
on `Docker Hub
215-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.7.1/images/sha256-63046938303e2770b7a6de2a94ffa46f7c9b805a84707b222272f54b8d0be781>`__.
215+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.8.0/images/sha256-e5c1e0058097ec16061c1412b3690600400492d89914c1698b886096c5159b6a>`__.
216216

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

219219
.. tab-set::
220220

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

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

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

240240
* - 24.04
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>`__
241+
- `1.7.0+rocm7.1.1.git7a57becf <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/>`__
242+
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
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.1_ubuntu24.04_py3.12_pytorch_release_2.6.0``
247+
``rocm/pytorch:rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.7.1``
248248
on `Docker Hub
249-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu24.04_py3.12_pytorch_release_2.6.0/images/sha256-40c2e5583ac79bfed54f44dd4e57f3a00d45199e4cba0850aace06361d5cea63>`__.
249+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.7.1/images/sha256-026884a87f9a79e43f2f39e7e3f69ddba3681d6e3c1bb26e0e48e648eba8724d>`__.
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.1_ubuntu22.04_py3.10_pytorch_release_2.6.0
257+
docker pull rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.7.1
258258
259259
.. rubric:: Additional software components
260260

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

270270
* - 22.04
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>`__
271+
- `1.7.0+rocm7.1.1.git7a57becf <https://repo.radeon.com/rocm/manylinux/rocm-rel-7.1.1/>`__
272+
- `0.22.1 <https://github.com/pytorch/vision/tree/v0.22.1>`__
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.1_ubuntu22.04_py3.10_pytorch_release_2.6.0``
277+
``rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.7.1``
278278
on `Docker Hub
279-
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1_ubuntu22.04_py3.10_pytorch_release_2.6.0/images/sha256-846520af9b22ce5aee1ddc9be7df3bec5257bf783ab8d084d33a0dcd48a430bf>`__.
279+
<https://hub.docker.com/layers/rocm/pytorch/rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.7.1/images/sha256-465c172cca7d6cfdd3504e8ed423ef6d6a427d8ad15924c377a7ee82247c9f63>`__.
280280

281281
.. _install_pytorch_wheels:
282282
.. _using-wheels-package:

0 commit comments

Comments
 (0)