Skip to content

Commit 0ce759c

Browse files
authored
Updated pytorch version in docker and older versions tests (#3200)
* Updated pytorch version in docker and older versions tests * Updated docker pth version
1 parent df819ca commit 0ce759c

File tree

2 files changed

+4
-14
lines changed

2 files changed

+4
-14
lines changed

.github/workflows/pytorch-version-tests.yml

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,30 +12,20 @@ jobs:
1212
runs-on: ubuntu-latest
1313
timeout-minutes: 45
1414
strategy:
15-
max-parallel: 7
15+
max-parallel: 5
1616
fail-fast: false
1717
matrix:
1818
python-version: [3.8, 3.9, "3.10"]
1919
pytorch-version:
20-
[2.0.1, 1.13.1, 1.12.1, 1.11.0, 1.10.0, 1.9.1, 1.8.1, 1.7.1, 1.6.0, 1.5.1]
20+
[2.1.2, 2.0.1, 1.13.1, 1.12.1, 1.11.0, 1.10.0, 1.9.1, 1.8.1, 1.5.1]
2121
exclude:
2222
- pytorch-version: 1.5.1
2323
python-version: 3.9
2424
- pytorch-version: 1.5.1
2525
python-version: "3.10"
2626

27-
- pytorch-version: 1.6.0
28-
python-version: 3.9
29-
- pytorch-version: 1.6.0
30-
python-version: "3.10"
31-
3227
# disabling python 3.9 support with PyTorch 1.7.1 and 1.8.1, to stop repeated pytorch-version test fail.
3328
# https://github.com/pytorch/ignite/issues/2383
34-
- pytorch-version: 1.7.1
35-
python-version: 3.9
36-
- pytorch-version: 1.7.1
37-
python-version: "3.10"
38-
3929
- pytorch-version: 1.8.1
4030
python-version: 3.9
4131
- pytorch-version: 1.8.1

docker/docker.cfg

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
[DEFAULT]
2-
build_docker_image_pytorch_version = 2.1.0-cuda12.1-cudnn8
2+
build_docker_image_pytorch_version = 2.2.1-cuda12.1-cudnn8
33
build_docker_image_hvd_version = v0.28.1
4-
build_docker_image_msdp_version = v0.8.1
4+
build_docker_image_msdp_version = v0.14.0

0 commit comments

Comments
 (0)