Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Upgrade CI to PyTorch 1.13 #15403

Merged
merged 46 commits into from
Nov 12, 2022
Merged

Upgrade CI to PyTorch 1.13 #15403

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
46 commits
Select commit Hold shift + click to select a range
cff2e96
requirements
Borda Oct 28, 2022
2874b8f
bump CI
Borda Oct 28, 2022
970d922
cleaning
Borda Oct 28, 2022
9feaac9
group-check
Borda Oct 28, 2022
6cb882a
docker
Borda Oct 28, 2022
6b37f1f
if
Borda Oct 31, 2022
074eb0b
Merge branch 'master' into ci/pt-1.13
Borda Nov 1, 2022
d6dedae
Merge branch 'master' into ci/pt-1.13
Borda Nov 1, 2022
be1bfc7
Merge branch 'master' into ci/pt-1.13
Borda Nov 1, 2022
d83eab7
update
Borda Nov 1, 2022
21ca4ae
push
Borda Nov 1, 2022
3ee95a2
py3.9
Borda Nov 2, 2022
41c0249
py3.10
Borda Nov 2, 2022
dc23fbe
drop bagua
Borda Nov 3, 2022
4885f99
checks
Borda Nov 4, 2022
10567dc
Merge branch 'master' into ci/pt-1.13
Borda Nov 4, 2022
8bec290
dockers
Borda Nov 5, 2022
3296496
Merge branch 'master' into ci/pt-1.13
akihironitta Nov 7, 2022
6a3961f
Merge branch 'master' into ci/pt-1.13
Borda Nov 8, 2022
def2885
Merge branch 'master' into ci/pt-1.13
akihironitta Nov 9, 2022
54516f7
Update checkgroup
akihironitta Nov 9, 2022
d092536
Skip 3.10 as there's no distribution of bagua and colossalai
akihironitta Nov 9, 2022
e1ca8b2
Apply suggestions from code review
Borda Nov 9, 2022
152b27d
Merge branch 'master' into ci/pt-1.13
Borda Nov 9, 2022
129453a
group
awaelchli Nov 8, 2022
2e3af53
rev
Borda Nov 9, 2022
99c1136
Merge branch 'master' into ci/pt-1.13
Borda Nov 10, 2022
e6d3870
.10
Borda Nov 10, 2022
34c1675
fix version issue in test
awaelchli Nov 10, 2022
47052a2
70min
Borda Nov 10, 2022
66e0ec9
Merge branch 'master' into ci/pt-1.13
Borda Nov 10, 2022
02538f4
Apply suggestions from code review
Borda Nov 11, 2022
55eefcf
Merge branch 'master' into ci/pt-1.13
Borda Nov 11, 2022
bb0a52d
Merge branch 'master' into ci/pt-1.13
Borda Nov 11, 2022
6271c04
Merge branch 'master' into ci/pt-1.13
Borda Nov 11, 2022
37d0d62
Merge branch 'master' into ci/pt-1.13
carmocca Nov 11, 2022
b8638c7
Remove stale comments
carmocca Nov 11, 2022
8f3d2c2
Merge branch 'master' into ci/pt-1.13
Borda Nov 11, 2022
39a9a43
Merge branch 'master' into ci/pt-1.13
Borda Nov 12, 2022
ea56ffc
Merge branch 'master' into ci/pt-1.13
carmocca Nov 12, 2022
6a52596
Unused changes
carmocca Nov 12, 2022
60a2701
Fix Lite tests according to current checkgroup config
carmocca Nov 12, 2022
1ad9c2a
Match PL tests config (+SLOW) to Lite test config
carmocca Nov 12, 2022
98bc516
TODO
carmocca Nov 12, 2022
bb94f8a
Merge branch 'master' into ci/pt-1.13
carmocca Nov 12, 2022
89601fb
Try 3.8
carmocca Nov 12, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
117 changes: 53 additions & 64 deletions .github/checkgroup.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,27 +25,25 @@ subprojects:
- ".actions/**"
checks:
# Note: updates here should be applied to the lightning_lite group
- "pl-cpu (macOS-11, pytorch, 3.8, 1.10)"
- "pl-cpu (macOS-11, pytorch, 3.9, 1.11)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.12)"
- "pl-cpu (macOS-11, pytorch, 3.8, 1.11)"
- "pl-cpu (macOS-11, pytorch, 3.9, 1.12)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.13)"
- "pl-cpu (macOS-11, pytorch, 3.7, 1.9, oldest)"
Borda marked this conversation as resolved.
Show resolved Hide resolved
- "pl-cpu (macOS-11, pytorch, 3.10, 1.13, pre)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.10)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.12)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.12)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.13)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.13, pre)"
- "pl-cpu (windows-2022, pytorch, 3.9, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.8, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.9, 1.12)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.12)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.13)"
- "pl-cpu (windows-2022, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (windows-2022, pytorch, 3.8, 1.13, pre)"
- "pl-cpu (macOS-11, lightning, 3.10, 1.12)"
- "pl-cpu (macOS-11, lightning, 3.10, 1.13)"
#- "pl-cpu (macOS-11, lightning, 3.7, 1.9, oldest)"
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.12)"
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.13)"
#- "pl-cpu (ubuntu-20.04, lightning, 3.7, 1.9, oldest)"
- "pl-cpu (windows-2022, lightning, 3.10, 1.12)"
- "pl-cpu (windows-2022, lightning, 3.10, 1.13)"
#- "pl-cpu (windows-2022, lightning, 3.7, 1.9, oldest)"
- "make-doctest (pytorch)"
- "make-html (pytorch)"
Expand All @@ -63,27 +61,25 @@ subprojects:
paths:
- ".github/workflows/ci-pytorch-tests.yml"
checks:
- "pl-cpu (macOS-11, pytorch, 3.8, 1.10)"
- "pl-cpu (macOS-11, pytorch, 3.9, 1.11)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.12)"
- "pl-cpu (macOS-11, pytorch, 3.8, 1.11)"
- "pl-cpu (macOS-11, pytorch, 3.9, 1.12)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.13)"
- "pl-cpu (macOS-11, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.13, pre)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.10)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.12)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.12)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.13)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.13, pre)"
- "pl-cpu (windows-2022, pytorch, 3.9, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.8, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.9, 1.12)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.12)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.13)"
- "pl-cpu (windows-2022, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (windows-2022, pytorch, 3.8, 1.13, pre)"
- "pl-cpu (macOS-11, lightning, 3.10, 1.12)"
- "pl-cpu (macOS-11, lightning, 3.10, 1.13)"
#- "pl-cpu (macOS-11, lightning, 3.7, 1.9, oldest)"
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.12)"
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.13)"
#- "pl-cpu (ubuntu-20.04, lightning, 3.7, 1.9, oldest)"
- "pl-cpu (windows-2022, lightning, 3.10, 1.12)"
- "pl-cpu (windows-2022, lightning, 3.10, 1.13)"
#- "pl-cpu (windows-2022, lightning, 3.7, 1.9, oldest)"

- id: "pytorch_lightning: Slow"
Expand Down Expand Up @@ -141,15 +137,15 @@ subprojects:
- "setup.py"
- ".actions/**"
checks:
- "build-cuda (3.9, 1.10, 11.3.1)"
- "build-cuda (3.9, 1.11, 11.3.1)"
- "build-cuda (3.9, 1.12, 11.6.1)"
- "build-cuda (3.10, 1.11, 11.3.1)"
- "build-cuda (3.10, 1.12, 11.6.1)"
- "build-cuda (3.10, 1.13, 11.6.1)"
- "build-hpu (1.5.0, 1.11.0)"
- "build-ipu (3.9, 1.9)"
- "build-NGC"
- "build-pl (3.9, 1.10, 11.3.1)"
- "build-pl (3.9, 1.11, 11.3.1)"
- "build-pl (3.9, 1.12, 11.6.1)"
- "build-pl (3.10, 1.11, 11.3.1)"
- "build-pl (3.10, 1.12, 11.6.1)"
- "build-pl (3.10, 1.13, 11.6.1)"
akihironitta marked this conversation as resolved.
Show resolved Hide resolved
- "build-xla (3.7, 1.12)"

# SECTION: lightning_lite
Expand All @@ -161,48 +157,44 @@ subprojects:
- "setup.cfg" # includes pytest config
- ".actions/**"
checks:
- "lite-cpu (macOS-11, lite, 3.9, 1.11)"
- "lite-cpu (macOS-11, lite, 3.8, 1.10)"
- "lite-cpu (macOS-11, lite, 3.10, 1.12)"
- "lite-cpu (macOS-11, lite, 3.10, 1.13, pre)"
- "lite-cpu (macOS-11, lite, 3.8, 1.11)"
- "lite-cpu (macOS-11, lite, 3.9, 1.12)"
- "lite-cpu (macOS-11, lite, 3.10, 1.13)"
- "lite-cpu (macOS-11, lite, 3.7, 1.9, oldest)"
- "lite-cpu (ubuntu-20.04, lite, 3.8, 1.11)"
- "lite-cpu (ubuntu-20.04, lite, 3.8, 1.10)"
- "lite-cpu (ubuntu-20.04, lite, 3.9, 1.11)"
- "lite-cpu (ubuntu-20.04, lite, 3.10, 1.12)"
- "lite-cpu (ubuntu-20.04, lite, 3.10, 1.13)"
- "lite-cpu (ubuntu-20.04, lite, 3.7, 1.9, oldest)"
- "lite-cpu (ubuntu-20.04, lite, 3.9, 1.13, pre)"
- "lite-cpu (windows-2022, lite, 3.8, 1.9)"
- "lite-cpu (windows-2022, lite, 3.9, 1.10)"
- "lite-cpu (windows-2022, lite, 3.10, 1.11)"
- "lite-cpu (windows-2022, lite, 3.9, 1.11)"
- "lite-cpu (windows-2022, lite, 3.10, 1.12)"
- "lite-cpu (windows-2022, lite, 3.10, 1.13)"
- "lite-cpu (windows-2022, lite, 3.7, 1.9, oldest)"
- "lite-cpu (windows-2022, lite, 3.8, 1.13, pre)"
- "lite-cpu (macOS-11, lightning, 3.8, 1.12)"
- "lite-cpu (ubuntu-20.04, lightning, 3.8, 1.12)"
- "lite-cpu (windows-2022, lightning, 3.8, 1.12)"
- "lite-cpu (macOS-11, lightning, 3.8, 1.13)"
- "lite-cpu (ubuntu-20.04, lightning, 3.8, 1.13)"
- "lite-cpu (windows-2022, lightning, 3.8, 1.13)"
- "lightning-lite (GPUs)"
- "mypy"
# Lite also requires PL checks as it depends on Lite
- "pl-cpu (macOS-11, pytorch, 3.8, 1.10)"
- "pl-cpu (macOS-11, pytorch, 3.9, 1.11)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.12)"
- "pl-cpu (macOS-11, pytorch, 3.8, 1.11)"
- "pl-cpu (macOS-11, pytorch, 3.9, 1.12)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.13)"
- "pl-cpu (macOS-11, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (macOS-11, pytorch, 3.10, 1.13, pre)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.10)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.8, 1.11)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.12)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.12)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.10, 1.13)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (ubuntu-20.04, pytorch, 3.9, 1.13, pre)"
- "pl-cpu (windows-2022, pytorch, 3.9, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.8, 1.11)"
- "pl-cpu (windows-2022, pytorch, 3.9, 1.12)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.12)"
- "pl-cpu (windows-2022, pytorch, 3.10, 1.13)"
- "pl-cpu (windows-2022, pytorch, 3.7, 1.9, oldest)"
- "pl-cpu (windows-2022, pytorch, 3.8, 1.13, pre)"
- "pl-cpu (macOS-11, lightning, 3.10, 1.12)"
- "pl-cpu (macOS-11, lightning, 3.10, 1.13)"
#- "pl-cpu (macOS-11, lightning, 3.7, 1.9, oldest)"
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.12)"
- "pl-cpu (ubuntu-20.04, lightning, 3.10, 1.13)"
#- "pl-cpu (ubuntu-20.04, lightning, 3.7, 1.9, oldest)"
- "pl-cpu (windows-2022, lightning, 3.10, 1.12)"
- "pl-cpu (windows-2022, lightning, 3.10, 1.13)"
#- "pl-cpu (windows-2022, lightning, 3.7, 1.9, oldest)"
- "make-doctest (pytorch)"
- "make-html (pytorch)"
Expand All @@ -221,18 +213,15 @@ subprojects:
- "lite-cpu (macOS-11, lite, 3.9, 1.11)"
- "lite-cpu (macOS-11, lite, 3.8, 1.10)"
- "lite-cpu (macOS-11, lite, 3.10, 1.12)"
- "lite-cpu (macOS-11, lite, 3.10, 1.13, pre)"
Borda marked this conversation as resolved.
Show resolved Hide resolved
- "lite-cpu (macOS-11, lite, 3.7, 1.9, oldest)"
- "lite-cpu (ubuntu-20.04, lite, 3.8, 1.11)"
- "lite-cpu (ubuntu-20.04, lite, 3.10, 1.12)"
- "lite-cpu (ubuntu-20.04, lite, 3.7, 1.9, oldest)"
- "lite-cpu (ubuntu-20.04, lite, 3.9, 1.13, pre)"
- "lite-cpu (windows-2022, lite, 3.8, 1.9)"
- "lite-cpu (windows-2022, lite, 3.9, 1.10)"
- "lite-cpu (windows-2022, lite, 3.10, 1.11)"
- "lite-cpu (windows-2022, lite, 3.10, 1.12)"
- "lite-cpu (windows-2022, lite, 3.7, 1.9, oldest)"
- "lite-cpu (windows-2022, lite, 3.8, 1.13, pre)"
- "lite-cpu (macOS-11, lightning, 3.8, 1.12)"
- "lite-cpu (ubuntu-20.04, lightning, 3.8, 1.12)"
- "lite-cpu (windows-2022, lightning, 3.8, 1.12)"
Expand Down
27 changes: 12 additions & 15 deletions .github/workflows/ci-lite-tests.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,28 +36,25 @@ jobs:
include:
# assign python and pytorch version combinations to operating systems (arbitrarily)
# note: there's no distribution of Torch==1.9 for Python>=3.9 or torch==1.10 for Python>=3.10
- {os: "macOS-11", pkg-name: "lite", python-version: "3.9", pytorch-version: "1.11"}
- {os: "macOS-11", pkg-name: "lite", python-version: "3.8", pytorch-version: "1.10"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.11"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.9", pytorch-version: "1.10"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.8", pytorch-version: "1.9"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.8", pytorch-version: "1.11"}
# only run PyTorch latest with Python latest
- {os: "macOS-11", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.12"}
- {os: "macOS-11", pkg-name: "lite", python-version: "3.8", pytorch-version: "1.11"}
- {os: "macOS-11", pkg-name: "lite", python-version: "3.9", pytorch-version: "1.12"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.8", pytorch-version: "1.10"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.9", pytorch-version: "1.11"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.12"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.9", pytorch-version: "1.11"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.12"}
# only run PyTorch latest with Python latest
- {os: "macOS-11", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.13"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.13"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.13"}
# "oldest" versions tests, only on minimum Python
- {os: "macOS-11", pkg-name: "lite", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
# release-candidate tests, mixed Python versions
- {os: "macOS-11", pkg-name: "lite", python-version: "3.10", pytorch-version: "1.13", release: "pre"}
- {os: "ubuntu-20.04", pkg-name: "lite", python-version: "3.9", pytorch-version: "1.13", release: "pre"}
- {os: "windows-2022", pkg-name: "lite", python-version: "3.8", pytorch-version: "1.13", release: "pre"}
# "lightning" installs the monolithic package
- {os: "macOS-11", pkg-name: "lightning", python-version: "3.8", pytorch-version: "1.12"}
- {os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.8", pytorch-version: "1.12"}
- {os: "windows-2022", pkg-name: "lightning", python-version: "3.8", pytorch-version: "1.12"}
- {os: "macOS-11", pkg-name: "lightning", python-version: "3.9", pytorch-version: "1.13"}
- {os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.9", pytorch-version: "1.13"}
- {os: "windows-2022", pkg-name: "lightning", python-version: "3.9", pytorch-version: "1.13"}

timeout-minutes: 15

Expand Down
12 changes: 6 additions & 6 deletions .github/workflows/ci-pytorch-dockers.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,9 @@ 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.10", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.10", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.10", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.10", pytorch_version: "1.13", cuda_version: "11.6.1"}
steps:
- uses: actions/checkout@v3
- uses: docker/setup-buildx-action@v2
Expand Down Expand Up @@ -93,9 +93,9 @@ 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.10", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.10", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.10", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.10", pytorch_version: "1.13", cuda_version: "11.6.1"}
steps:
- uses: actions/checkout@v3
- uses: docker/setup-buildx-action@v2
Expand Down
30 changes: 13 additions & 17 deletions .github/workflows/ci-pytorch-tests.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,32 +40,28 @@ jobs:
include:
# assign python and pytorch version combinations to operating systems (arbitrarily)
# note: there's no distribution of Torch==1.9 for Python>=3.9 or torch==1.10 for Python>=3.10
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.11"}
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.10"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.10"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.11"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.11"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.10"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.11"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.11"}
# only run PyTorch latest with Python latest
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.12"}
- {os: "macOS-11", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.12"}
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.11"}
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.12"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.11"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.12"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.12"}
- {os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.12"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.11"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.12"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.12"}
- {os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.12"}
# only run PyTorch latest with Python latest
akihironitta marked this conversation as resolved.
Show resolved Hide resolved
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.13"}
- {os: "macOS-11", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.13"}
- {os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.13"}
- {os: "windows-2022", pkg-name: "lightning", python-version: "3.10", pytorch-version: "1.13"}
# "oldest" versions tests, only on minimum Python
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "macOS-11", pkg-name: "lightning", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "ubuntu-20.04", pkg-name: "lightning", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
- {os: "windows-2022", pkg-name: "lightning", python-version: "3.7", pytorch-version: "1.9", requires: "oldest"}
# release-candidate tests, mixed Python versions
- {os: "macOS-11", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "1.13", release: "pre"}
- {os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "1.13", release: "pre"}
- {os: "windows-2022", pkg-name: "pytorch", python-version: "3.8", pytorch-version: "1.13", release: "pre"}

timeout-minutes: 60

Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/release-docker.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,9 @@ jobs:
matrix:
include:
# We only release one docker image per PyTorch version.
- {python_version: "3.9", pytorch_version: "1.10", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.9", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.10", pytorch_version: "1.11", cuda_version: "11.3.1"}
- {python_version: "3.10", pytorch_version: "1.12", cuda_version: "11.6.1"}
- {python_version: "3.10", pytorch_version: "1.13", cuda_version: "11.6.1"}
steps:
- name: Checkout
uses: actions/checkout@v3
Expand Down
4 changes: 3 additions & 1 deletion dockers/base-cuda/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,9 @@ RUN \
CUDA_VERSION_MM=$(python -c "print(''.join('$CUDA_VERSION'.split('.')[:2]))") && \
CUDA_VERSION_BAGUA=$(python -c "print([ver for ver in [116,113,111,102] if $CUDA_VERSION_MM >= ver][0])") && \
pip install "bagua-cuda$CUDA_VERSION_BAGUA" && \
if [[ "$CUDA_VERSION_MM" = "$CUDA_VERSION_BAGUA" ]]; then python -c "import bagua_core; bagua_core.install_deps()"; fi && \
if [[ "$CUDA_VERSION_MM" = "$CUDA_VERSION_BAGUA" ]]; then \
python -c "import bagua_core; bagua_core.install_deps()"; \
fi && \
python -c "import bagua; print(bagua.__version__)"

RUN \
Expand Down
Loading