From 2c85529bfc7bd0068f2395188bbf54c9adb01422 Mon Sep 17 00:00:00 2001 From: Siyuan Liu Date: Thu, 23 Jan 2025 10:50:16 -0800 Subject: [PATCH] [TPU] Update TPU CI to use torchxla nightly on 20250122 (#12334) Signed-off-by: Siyuan Liu --- Dockerfile.tpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.tpu b/Dockerfile.tpu index b617932a85b47..ee0d94d98e82b 100644 --- a/Dockerfile.tpu +++ b/Dockerfile.tpu @@ -1,4 +1,4 @@ -ARG NIGHTLY_DATE="20241017" +ARG NIGHTLY_DATE="20250122" ARG BASE_IMAGE="us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:nightly_3.10_tpuvm_$NIGHTLY_DATE" FROM $BASE_IMAGE