From e22dc297279125dad577b19e938aba18b8effb55 Mon Sep 17 00:00:00 2001 From: Lucas Manning Date: Mon, 11 Nov 2024 16:07:22 -0800 Subject: [PATCH] Update pytorch image python module versions. The switch to a newer numpy was breaking guitar tests. PiperOrigin-RevId: 695514200 --- images/tpu/pytorch/Dockerfile.x86_64 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/images/tpu/pytorch/Dockerfile.x86_64 b/images/tpu/pytorch/Dockerfile.x86_64 index d78be8816..1ffc07bec 100644 --- a/images/tpu/pytorch/Dockerfile.x86_64 +++ b/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 \ No newline at end of file