ci: bump CUDA for nightly (#18162)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
This commit is contained in:
Jirka Borovec
GitHub
parent
4c57c0bc07
commit
14b65941b0
|
|
@ -46,7 +46,7 @@ jobs:
|
|||
variables:
|
||||
DEVICES: $( python -c 'print("$(Agent.Name)".split("_")[-1])' )
|
||||
container:
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
options: "--gpus=all --shm-size=32g"
|
||||
strategy:
|
||||
matrix:
|
||||
|
|
|
|||
|
|
@ -60,15 +60,15 @@ jobs:
|
|||
strategy:
|
||||
matrix:
|
||||
'Fabric | latest':
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
IS_NIGHTLY: "false"
|
||||
PACKAGE_NAME: "fabric"
|
||||
'Lightning | latest':
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
IS_NIGHTLY: "false"
|
||||
PACKAGE_NAME: "lightning"
|
||||
'Lightning | nightly':
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
IS_NIGHTLY: "true"
|
||||
PACKAGE_NAME: "lightning"
|
||||
workspace:
|
||||
|
|
|
|||
|
|
@ -52,15 +52,15 @@ jobs:
|
|||
strategy:
|
||||
matrix:
|
||||
'PyTorch | latest':
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
IS_NIGHTLY: "false"
|
||||
PACKAGE_NAME: "pytorch"
|
||||
'Lightning | latest':
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
IS_NIGHTLY: "false"
|
||||
PACKAGE_NAME: "lightning"
|
||||
'Lightning | nightly':
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.7.1"
|
||||
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
|
||||
IS_NIGHTLY: "true"
|
||||
PACKAGE_NAME: "lightning"
|
||||
pool: lit-rtx-3090
|
||||
|
|
|
|||
|
|
@ -139,15 +139,16 @@ subprojects:
|
|||
- "!*.md"
|
||||
- "!**/*.md"
|
||||
checks:
|
||||
- "build-cuda (3.9, 1.11, 11.3.1)"
|
||||
- "build-cuda (3.9, 1.12, 11.7.1)"
|
||||
- "build-cuda (3.9, 1.13, 11.8.0)"
|
||||
- "build-cuda (3.9, 1.13, 12.0.1)"
|
||||
- "build-cuda (3.10, 2.0, 11.8.0)"
|
||||
- "build-cuda (3.10, 2.0, 12.0.1)"
|
||||
- "build-cuda (3.10, 2.0, 11.7.1)"
|
||||
#- "build-NGC"
|
||||
- "build-pl (3.9, 1.11, 11.3.1)"
|
||||
- "build-pl (3.9, 1.12, 11.7.1)"
|
||||
- "build-pl (3.9, 1.13, 11.8.0)"
|
||||
- "build-pl (3.9, 1.13, 12.0.1)"
|
||||
- "build-pl (3.10, 2.0, 11.8.0)"
|
||||
- "build-pl (3.10, 2.0, 12.0.1)"
|
||||
|
||||
# SECTIONS: lightning_data
|
||||
|
|
|
|||
|
|
@ -42,9 +42,10 @@ jobs:
|
|||
include:
|
||||
# We only release one docker image per PyTorch version.
|
||||
# The matrix here is the same as the one in release-docker.yml.
|
||||
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
|
||||
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.7.1"}
|
||||
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "11.8.0"}
|
||||
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "12.0.1"}
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "11.8.0"}
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "12.0.1"}
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
|
|
@ -70,12 +71,11 @@ jobs:
|
|||
include:
|
||||
# These are the base images for PL release docker images,
|
||||
# so include at least all of the combinations in release-dockers.yml.
|
||||
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
|
||||
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.7.1"}
|
||||
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "11.8.0"}
|
||||
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "12.0.1"}
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "11.8.0"}
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "12.0.1"}
|
||||
# these are used in Azure GPU CI
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "11.7.1"}
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
- uses: docker/setup-buildx-action@v2
|
||||
|
|
|
|||
|
|
@ -16,9 +16,11 @@ jobs:
|
|||
matrix:
|
||||
include:
|
||||
# We only release one docker image per PyTorch version.
|
||||
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
|
||||
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.7.1"}
|
||||
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "11.8.0"}
|
||||
- {python_version: "3.9", pytorch_version: "1.13", cuda_version: "12.0.1"}
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "11.8.0"}
|
||||
- {python_version: "3.10", pytorch_version: "2.0", cuda_version: "12.0.1"}
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v3
|
||||
|
|
|
|||
Loading…
Reference in New Issue