Skip to content

Commit

Permalink
adding note to PyTorch docker install to highlight change in definiti…
Browse files Browse the repository at this point in the history
…on of latest
  • Loading branch information
yhuiYH committed Sep 13, 2024
1 parent f11a30b commit b0031ae
Showing 1 changed file with 18 additions and 1 deletion.
19 changes: 18 additions & 1 deletion docs/install/3rd-party/pytorch-install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,10 @@ To install PyTorch for ROCm, you have the following options:
* :ref:`using-pytorch-rocm-docker-image`
* :ref:`using-pytorch-upstream-docker-image`

.. |br| raw:: html

<br/>

For hardware, software, and third-party framework compatibility between ROCm and PyTorch, refer to:

* :ref:`system-requirements`
Expand All @@ -36,7 +40,20 @@ Using a Docker image with PyTorch pre-installed
docker pull rocm/pytorch:latest
You can also download a specific and supported configuration with different user-space ROCm
versions, PyTorch versions, and operating systems.
versions, PyTorch versions, and operating systems by filtering through the `tags <https://hub.docker.com/r/rocm/pytorch/tags>`_.

.. important::

As of ROCm 6.2.1, rocm/pytorch:latest will now point to the latest tested release version of PyTorch (e.g. PyTorch 2.3), similar to rocm/pytorch:latest-release tag.
Prior to ROCm 6.2.1, rocm/pytorch:latest was pointing to a development version of PyTorch, which did not correspond to any specific PyTorch release.

.. csv-table::
:header: "Description", "6.2.1 and newer", "6.2.0 and older"

"Latest tested *release* of PyTorch", **rocm/pytorch:latest** |br| rocm/pytorch:latest-release, rocm/pytorch:latest-release
"Latest dev version of PyTorch", rocm/pytorch:latest-internal, **rocm/pytorch:latest**



2. Start a Docker container using the image.

Expand Down

0 comments on commit b0031ae

Please sign in to comment.