From 8d165f58baa8d6aa08004381e707d1e8e3664dec Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Wed, 4 Sep 2024 12:45:54 -0700 Subject: [PATCH 1/2] Update upstream docker image to Python 3.10 --- .github/upstream/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/upstream/Dockerfile b/.github/upstream/Dockerfile index 1707cd4f70c1..1cd19e7d66d7 100644 --- a/.github/upstream/Dockerfile +++ b/.github/upstream/Dockerfile @@ -1,10 +1,10 @@ # Dockerfile for image used by upstream CI # This requires cuda & cudnn packages pre-installed in the base image. # Other available cuda images are listed at https://hub.docker.com/r/nvidia/cuda -ARG base_image="us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.8_cuda_12.1" +ARG base_image="us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.10_cuda_12.1" FROM "${base_image}" -ARG python_version="3.8" +ARG python_version="3.10" ARG cuda="1" ARG cuda_compute="5.2,7.5" ARG cc="clang" From a3ca019bfcb7b77818d24f22e00bf466668179f5 Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Wed, 4 Sep 2024 12:46:32 -0700 Subject: [PATCH 2/2] bump upstream docker image tag --- .github/workflows/build_upstream_image.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/build_upstream_image.yml b/.github/workflows/build_upstream_image.yml index 446ad366e547..ded8f07b7132 100644 --- a/.github/workflows/build_upstream_image.yml +++ b/.github/workflows/build_upstream_image.yml @@ -27,7 +27,7 @@ jobs: - name: Build Docker image shell: bash run: | - docker build -t "${ECR_DOCKER_IMAGE_BASE}:v1.2-lite" .github/upstream + docker build -t "${ECR_DOCKER_IMAGE_BASE}:v1.3-lite" .github/upstream - name: Stage image to ECR shell: bash run: |