diff --git a/docs/install/3rd-party/pytorch-install.rst b/docs/install/3rd-party/pytorch-install.rst index 84106bb9..2126462e 100644 --- a/docs/install/3rd-party/pytorch-install.rst +++ b/docs/install/3rd-party/pytorch-install.rst @@ -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.