From 2de89a7ddc5240ae58cc651a4030519cc44475b7 Mon Sep 17 00:00:00 2001 From: Simeet Nayan Date: Tue, 30 Jul 2024 23:40:38 +0530 Subject: [PATCH 1/2] update tpu-tests.yml with pip<24 --- .github/workflows/tpu-tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/tpu-tests.yml b/.github/workflows/tpu-tests.yml index ab14ad3c1de..85cd3ea96ad 100644 --- a/.github/workflows/tpu-tests.yml +++ b/.github/workflows/tpu-tests.yml @@ -50,7 +50,7 @@ jobs: - name: Get pip cache dir id: pip-cache run: | - pip3 install -U pip + pip3 install -U "pip<24" echo "pip_cache=$(pip cache dir)" >> $GITHUB_OUTPUT shell: bash -l {0} From 849ad855a03dffdcdb16a3be130bae1d09bf2b62 Mon Sep 17 00:00:00 2001 From: Simeet Nayan Date: Tue, 30 Jul 2024 23:49:15 +0530 Subject: [PATCH 2/2] update tpu-tests.yml --- .github/workflows/tpu-tests.yml | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/.github/workflows/tpu-tests.yml b/.github/workflows/tpu-tests.yml index 85cd3ea96ad..9002a7559b0 100644 --- a/.github/workflows/tpu-tests.yml +++ b/.github/workflows/tpu-tests.yml @@ -70,9 +70,8 @@ jobs: pip install mkl==2021.4.0 ## Install torch & xla and torchvision - pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch-nightly-cp39-cp39-linux_x86_64.whl - pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torch_xla-nightly-cp39-cp39-linux_x86_64.whl - pip install --pre https://storage.googleapis.com/tpu-pytorch/wheels/colab/torchvision-nightly-cp39-cp39-linux_x86_64.whl + pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu + pip install https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly-cp310-cp310-linux_x86_64.whl # Check installation python -c "import torch"