From 65352ad9d40c0ab614137110cb3719a1253c610e Mon Sep 17 00:00:00 2001 From: vfdev Date: Wed, 31 Jul 2024 05:53:18 +0200 Subject: [PATCH] Fix failing tpu tests (#3261) * Update tpu-tests.yml * Update tpu-tests.yml --- .github/workflows/tpu-tests.yml | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/.github/workflows/tpu-tests.yml b/.github/workflows/tpu-tests.yml index ab14ad3c1de..cc330de0c27 100644 --- a/.github/workflows/tpu-tests.yml +++ b/.github/workflows/tpu-tests.yml @@ -36,10 +36,10 @@ jobs: steps: - uses: actions/checkout@v4 - - name: Set up Python 3.9 + - name: Set up Python 3.10 uses: actions/setup-python@v4 with: - python-version: "3.9" + python-version: "3.10" architecture: "x64" - name: Get year & week number @@ -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} @@ -70,10 +70,9 @@ 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"