For information on available Radeon Software for Linux releases, please refer to Linux® Drivers for AMD Radeon™ and Radeon PRO™ Graphics.
#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.