Update pytorch image python module versions.

The switch to a newer numpy was breaking guitar tests.

PiperOrigin-RevId: 695514200

Author: manninglucas

images/tpu/pytorch/Dockerfile.x86_64

@@ -1,5 +1,5 @@
 FROM python:3.10
 
-RUN pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 torchvision -f https://storage.googleapis.com/libtpu-releases/index.html
+RUN pip install torch==2.5.1 torch_xla[tpu]==2.5.1 -f https://storage.googleapis.com/libtpu-releases/index.html
 
 RUN git clone --depth=1 --branch v2.0.0 https://github.com/pytorch/xla.git