for rocm5.5 compute platform index, use --index-url https://sonatype-nexus-community.github.io/pytorch-pypi/whl/rocm5.5/simple
--index-url https://sonatype-nexus-community.github.io/pytorch-pypi/whl/rocm5.5/simple
see also other available indexes