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