#For PyTorch 1.11 + torchvision 0.12 + RCCL 2.12
pip3 install torch==1.11.0 torchvision==0.12.0 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-5.2/rccl2.12/

#For PyTorch 1.12 + torchvision 0.13
pip3 install torch==1.12.0 torchvision==0.13.0 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-5.2/

#For PyTorch 1.11 + torchvision 0.12
pip3 install torch==1.11.0 torchvision==0.12.0 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-5.2/

#For PyTorch 1.10 + torchvision 0.11
pip3 install torch==1.10.0 torchvision==0.11.0 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-5.2/

------------------------------------------------------------
the .whl files are tested in rocm/dev-ubuntu-20.04:5.2 base docker image