Skip to content

Commit

Permalink
Update to ROCm6.2 pytorch nightly wheels (ROCm#229)
Browse files Browse the repository at this point in the history
  • Loading branch information
jithunnair-amd authored and spolifroni-amd committed Aug 28, 2024
1 parent ad2b9aa commit 051ed1c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/install/3rd-party/pytorch-install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -128,12 +128,12 @@ wheels command, you must select 'Linux', 'Python', 'pip', and 'ROCm' in the matr

.. note::

The following command uses the ROCm 6.1 PyTorch wheel. If you want a different version of ROCm,
The following command uses the ROCm 6.2 PyTorch wheel. If you want a different version of ROCm,
modify the command accordingly.

.. code-block:: bash
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1/
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2/
4. (Optional) Use MIOpen kdb files with ROCm PyTorch wheels.

Expand Down

0 comments on commit 051ed1c

Please sign in to comment.