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}"