Skip to content

Commit c7c42dc

Browse files
awaelchliawaelchli
authored
Update CI to use latest PyTorch 2.2 (#19401)
1 parent fb0ce03 commit c7c42dc

File tree

6 files changed

+7
-25
lines changed

6 files changed

+7
-25
lines changed

‎.azure/gpu-benchmarks.yml‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,8 @@ jobs:
4646
variables:
4747
DEVICES: $( python -c 'print("$(Agent.Name)".split("_")[-1])' )
4848
container:
49-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.0-cuda11.8.0"
49+
# TODO: Upgrade to Python 3.11
50+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.2-cuda12.1.0"
5051
options: "--gpus=all --shm-size=32g"
5152
strategy:
5253
matrix:

‎.azure/gpu-tests-fabric.yml‎

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,14 +56,12 @@ jobs:
5656
options: "--gpus=all --shm-size=2gb -v /var/tmp:/var/tmp"
5757
strategy:
5858
matrix:
59+
# TODO: Upgrade to Python 3.11
5960
"Fabric | latest":
60-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.0"
61-
PACKAGE_NAME: "fabric"
62-
"Fabric | future":
63-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.2-cuda12.1.0"
61+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.2-cuda12.1.0"
6462
PACKAGE_NAME: "fabric"
6563
"Lightning | latest":
66-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.0"
64+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.2-cuda12.1.0"
6765
PACKAGE_NAME: "lightning"
6866
workspace:
6967
clean: all

‎.azure/gpu-tests-pytorch.yml‎

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -48,15 +48,12 @@ jobs:
4848
cancelTimeoutInMinutes: "2"
4949
strategy:
5050
matrix:
51+
# TODO: Upgrade to Python 3.11
5152
"PyTorch | latest":
52-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.0"
53-
PACKAGE_NAME: "pytorch"
54-
"PyTorch | future":
55-
# todo: failed to install `pygame` with py3.11
5653
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.2-cuda12.1.0"
5754
PACKAGE_NAME: "pytorch"
5855
"Lightning | latest":
59-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.0"
56+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.2-cuda12.1.0"
6057
PACKAGE_NAME: "lightning"
6158
pool: lit-rtx-3090
6259
variables:

‎.github/checkgroup.yml‎

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,12 @@ subprojects:
2121
checks:
2222
- "pl-cpu (macOS-11, lightning, 3.8, 1.13, oldest)"
2323
- "pl-cpu (macOS-11, lightning, 3.10, 1.13)"
24-
- "pl-cpu (macOS-11, lightning, 3.10, 2.0)"
2524
- "pl-cpu (macOS-11, lightning, 3.10, 2.1)"
2625
- "pl-cpu (ubuntu-20.04, lightning, 3.8, 1.13, oldest)"
2726
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.13)"
28-
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 2.0)"
2927
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 2.1)"
3028
- "pl-cpu (windows-2022, lightning, 3.8, 1.13, oldest)"
3129
- "pl-cpu (windows-2022, lightning, 3.10, 1.13)"
32-
- "pl-cpu (windows-2022, lightning, 3.10, 2.0)"
3330
- "pl-cpu (windows-2022, lightning, 3.10, 2.1)"
3431
- "pl-cpu (macOS-11, pytorch, 3.8, 1.13)"
3532
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.13)"
@@ -190,15 +187,12 @@ subprojects:
190187
checks:
191188
- "fabric-cpu (macOS-11, lightning, 3.8, 1.13, oldest)"
192189
- "fabric-cpu (macOS-11, lightning, 3.10, 1.13)"
193-
- "fabric-cpu (macOS-11, lightning, 3.10, 2.0)"
194190
- "fabric-cpu (macOS-11, lightning, 3.11, 2.1)"
195191
- "fabric-cpu (ubuntu-20.04, lightning, 3.8, 1.13, oldest)"
196192
- "fabric-cpu (ubuntu-20.04, lightning, 3.10, 1.13)"
197-
- "fabric-cpu (ubuntu-20.04, lightning, 3.10, 2.0)"
198193
- "fabric-cpu (ubuntu-20.04, lightning, 3.11, 2.1)"
199194
- "fabric-cpu (windows-2022, lightning, 3.8, 1.13, oldest)"
200195
- "fabric-cpu (windows-2022, lightning, 3.10, 1.13)"
201-
- "fabric-cpu (windows-2022, lightning, 3.10, 2.0)"
202196
- "fabric-cpu (windows-2022, lightning, 3.11, 2.1)"
203197
- "fabric-cpu (macOS-11, fabric, 3.8, 1.13)"
204198
- "fabric-cpu (ubuntu-20.04, fabric, 3.8, 1.13)"

‎.github/workflows/ci-tests-fabric.yml‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,13 +43,9 @@ jobs:
4343
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13" }
4444
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13" }
4545
# only run PyTorch latest
46-
- { os: "macOS-11", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.0" }
47-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.0" }
48-
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.0" }
4946
- { os: "macOS-11", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.1" }
5047
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.1" }
5148
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.1" }
52-
# only run PyTorch future
5349
- { os: "macOS-12", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2" }
5450
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2" }
5551
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.11", pytorch-version: "2.2" }

‎.github/workflows/ci-tests-pytorch.yml‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,13 +47,9 @@ jobs:
4747
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13" }
4848
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13" }
4949
# only run PyTorch latest
50-
- { os: "macOS-11", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.0" }
51-
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.0" }
52-
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.0" }
5350
- { os: "macOS-11", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
5451
- { os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
5552
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.1" }
56-
# only run PyTorch future
5753
- { os: "macOS-12", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.2" }
5854
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.2" }
5955
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "2.2" }

0 commit comments

Comments
 (0)