Pin pytorch in ROCM dockerfile

Author: Mark Saroufim

docker/amd-docker.Dockerfile

@@ -40,7 +40,7 @@ RUN sudo apt update -y \
 
 RUN sudo pip install --upgrade pip
 
-RUN sudo pip install --no-cache-dir pytorch-triton-rocm torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3
+RUN sudo pip install --no-cache-dir torch==2.10.0.dev20250916+rocm6.3 pytorch-triton-rocm --index-url https://download.pytorch.org/whl/nightly/rocm6.3
 
 RUN git clone --recursive https://github.com/ROCm/aiter.git \
     && cd aiter \