Skip to content

Commit 9bd0844

Browse files
committed
Update docker images to build
1 parent 4e9fbdb commit 9bd0844

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

.github/workflows/cicd-pytorch_dockers.yml

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@ jobs:
3030
fail-fast: false
3131
matrix:
3232
# the config used in '.azure-pipelines/gpu-tests.yml' since the Dockerfile uses the cuda image
33-
python_version: ["3.9"]
34-
pytorch_version: ["1.10", "1.11"]
33+
python_version: ["3.10"]
34+
pytorch_version: ["1.11", "1.12"]
3535
steps:
3636
- uses: actions/checkout@v2
3737
- uses: docker/setup-buildx-action@v2
@@ -50,8 +50,8 @@ jobs:
5050
fail-fast: false
5151
matrix:
5252
# the config used in '.circleci/config.yml`'
53-
python_version: ["3.7"]
54-
xla_version: ["1.11"]
53+
python_version: ["3.9", "3.10"]
54+
xla_version: ["1.12"]
5555
steps:
5656
- uses: actions/checkout@v2
5757
- uses: docker/setup-buildx-action@v2
@@ -86,12 +86,11 @@ jobs:
8686
matrix:
8787
include:
8888
# the config used in '.azure-pipelines/gpu-tests.yml'
89-
- {python_version: "3.7", pytorch_version: "1.10", cuda_version: "11.1", ubuntu_version: "20.04"}
90-
- {python_version: "3.7", pytorch_version: "1.11", cuda_version: "11.3.1", ubuntu_version: "20.04"}
91-
# latest (used in Tutorials)
89+
# Some are used in different projects such as tutorials.
9290
- {python_version: "3.8", pytorch_version: "1.9", cuda_version: "11.1", ubuntu_version: "20.04"}
9391
- {python_version: "3.9", pytorch_version: "1.10", cuda_version: "11.1", ubuntu_version: "20.04"}
9492
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1", ubuntu_version: "20.04"}
93+
- {python_version: "3.10", pytorch_version: "1.12", cuda_version: "11.6.1", ubuntu_version: "20.04"}
9594
steps:
9695
- uses: actions/checkout@v2
9796
- uses: docker/setup-buildx-action@v2
@@ -130,8 +129,7 @@ jobs:
130129
- {python_version: "3.8", pytorch_version: "1.9", cuda_version: "11.1"}
131130
- {python_version: "3.8", pytorch_version: "1.10", cuda_version: "11.1"}
132131
- {python_version: "3.9", pytorch_version: "1.11", cuda_version: "11.3.1"}
133-
# nightly: add when there's a release candidate
134-
# - {python_version: "3.9", pytorch_version: "1.12"}
132+
- {python_version: "3.10", pytorch_version: "1.12", cuda_version: "11.6.1"}
135133
steps:
136134
- uses: actions/checkout@v2
137135
- uses: docker/setup-buildx-action@v2

0 commit comments

Comments
 (0)