Dockerfile.rocm.ubi: bump pytorch nightly version

Author: dtrifiro

Dockerfile.rocm.ubi

@@ -49,8 +49,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
     uv pip install --pre \
         --index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \
-        torch==2.6.0.dev20250104+rocm${version}\
-        torchvision==0.22.0.dev20250104+rocm${version} && \
+        torch==2.7.0.dev20250308+rocm${version}\
+        torchvision==0.22.0.dev20250308+rocm${version} && \
     # Install libdrm-amdgpu to avoid errors when retrieving device information (amdgpu.ids: No such file or directory)
     microdnf install -y libdrm-amdgpu && \
     microdnf clean all