Switch GPU CI from nightly to RC/test (#18485)

This commit is contained in:
Jirka Borovec 2023-09-06 23:34:43 +02:00 committed by GitHub
parent cf437ed7c6
commit eb8b314f7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions

View File

@ -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

View File

@ -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