Release notes

For information on available ROCm releases, please refer to the ROCm Release Notes

For information on available Radeon Software for Linux releases, please refer to Linux® Drivers for AMD Radeon™ and Radeon PRO™ Graphics.

ROCm PyTorch wheel install instructions

#For PyTorch 2.2.1 + torchvision 0.17.1
pip3 install torch==2.2.1 torchvision==0.17.1 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/preview/

#For PyTorch 2.3.0 + torchvision 0.18.0
pip3 install torch==2.3.0 torchvision==0.18.0 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1/preview/

-------------------------------------------------------------------------------------------
Note:
    These wheels are available for community use with the latest PyTorch versions on ROCm6.1, but have not been fully validated by the ROCm QA team.
    A compatible pytorch-triton-rocm package will also be installed as a dependency when the version of PyTorch wheels is greater than 2.0.


Using MIOpen kdb files with ROCm PyTorch wheels

Check README