diff --git a/infra/ansible/ptxla_docker_for_torchprime.Dockerfile b/infra/ansible/ptxla_docker_for_torchprime.Dockerfile
index 6de990661526..e9885bd49284 100644
--- a/infra/ansible/ptxla_docker_for_torchprime.Dockerfile
+++ b/infra/ansible/ptxla_docker_for_torchprime.Dockerfile
@@ -23,7 +23,7 @@
 #
 # (Googlers only) Refer to go/ptxla-torchprime-trigger for information on retention policy of
 # the docker images.
-ARG python_version=3.10
+ARG python_version=3.12
 ARG debian_version=bullseye
 
 FROM python:${python_version}-${debian_version} AS release
diff --git a/infra/ansible/publish_torchprime_e2e_test_docker.sh b/infra/ansible/publish_torchprime_e2e_test_docker.sh
index 4e8c3d23009d..cee169b95d79 100755
--- a/infra/ansible/publish_torchprime_e2e_test_docker.sh
+++ b/infra/ansible/publish_torchprime_e2e_test_docker.sh
@@ -66,7 +66,7 @@ docker build -t "${DOCKER_URL}" \
     "${DEFAULT_CONTEXT_PATH}" \
     -f "${DOCKERFILE_PATH}" \
     --build-arg ansible_vars="${ANSIBLE_VARS_COMPACT}" \
-    --build-arg python_version=3.10 \
+    --build-arg python_version=3.12 \
     --build-arg debian_version=bullseye
 docker push "${DOCKER_URL}"