#For PyTorch 1.12.1 + torchvision 0.13.1
pip3 install torch==1.12.1 torchvision==0.13.1 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-5.3/

#For PyTorch 1.10.1 + torchvision 0.11.2
pip3 install torch==1.10.1 torchvision==0.11.2 -f https://repo.radeon.com/rocm/manylinux/rocm-rel-5.3/

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