Skip to content

Commit

Permalink
Update to ROCm6.2 pytorch nightly wheels (ROCm#229) (ROCm#281)
Browse files Browse the repository at this point in the history
Co-authored-by: Jithun Nair <[email protected]>
  • Loading branch information
spolifroni-amd and jithunnair-amd committed Aug 29, 2024
1 parent b0329fb commit aab3fcf
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 aab3fcf

Please sign in to comment.