From eb8b314f7cbb5e87e0d149f95f45f1409a7715c2 Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Wed, 6 Sep 2023 23:34:43 +0200 Subject: [PATCH] Switch GPU CI from nightly to RC/test (#18485) --- .azure/gpu-tests-fabric.yml | 6 +++--- .azure/gpu-tests-pytorch.yml | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml index 6ff436a29a..340a076efc 100644 --- a/.azure/gpu-tests-fabric.yml +++ b/.azure/gpu-tests-fabric.yml @@ -66,7 +66,7 @@ jobs: image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0" IS_NIGHTLY: "false" PACKAGE_NAME: "lightning" - 'Lightning | nightly': + 'Lightning | RC': image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0" IS_NIGHTLY: "true" PACKAGE_NAME: "lightning" @@ -115,11 +115,11 @@ jobs: - bash: | pip uninstall -y torch torchvision - pip install torch torchvision -U --pre --no-cache --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION_MM%} + pip install torch torchvision -U --pre --no-cache --index-url https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM%} python -c "from torch import __version__ as ver; assert ver.startswith('2.1.0'), ver" # without succeeded this could run even if the job has already failed condition: and(succeeded(), eq(variables.IS_NIGHTLY, 'true')) - displayName: 'Bump to nightly' + displayName: 'Bump to RC' - bash: | set -e diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml index 1809a9da10..da5914b28a 100644 --- a/.azure/gpu-tests-pytorch.yml +++ b/.azure/gpu-tests-pytorch.yml @@ -58,7 +58,7 @@ jobs: image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0" IS_NIGHTLY: "false" PACKAGE_NAME: "lightning" - 'Lightning | nightly': + 'Lightning | RC': image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0" IS_NIGHTLY: "true" PACKAGE_NAME: "lightning" @@ -125,11 +125,11 @@ jobs: - bash: | pip uninstall -y torch torchvision - pip install torch torchvision -U --pre --no-cache --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION_MM%} + pip install torch torchvision -U --pre --no-cache --index-url https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM%} python -c "from torch import __version__ as ver; assert ver.startswith('2.1.0'), ver" # without succeeded this could run even if the job has already failed condition: and(succeeded(), eq(variables.IS_NIGHTLY, 'true')) - displayName: 'Bump to nightly' + displayName: 'Bump to RC' - bash: pip uninstall -y lightning # without succeeded this could run even if the job has already failed