Switch GPU CI from nightly to RC/test (#18485)
This commit is contained in:
parent
cf437ed7c6
commit
eb8b314f7c
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue