Dockerfile.rocm.ubi: fix torch extra index url

Author: dtrifiro

Dockerfile.rocm.ubi

@@ -45,7 +45,7 @@ gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key" > /etc/yum.repos.d/amdgpu.repo
 RUN --mount=type=cache,target=/root/.cache/uv \
     export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
     uv pip install --pre \
-        --index-url "https://download.pytorch.org/whl/ \
+        --index-url "https://download.pytorch.org/whl/rocm${version}" \
         torch==2.7.0+rocm${version}\
         torchvision==0.22.0+rocm${version} && \
     # Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)