Skip to content

Commit

Permalink
Updated pytorch version in docker and older versions tests
Browse files Browse the repository at this point in the history
  • Loading branch information
vfdev-5 authored Feb 19, 2024
1 parent 72a3c30 commit cf26025
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 13 deletions.
14 changes: 2 additions & 12 deletions .github/workflows/pytorch-version-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,30 +12,20 @@ jobs:
runs-on: ubuntu-latest
timeout-minutes: 45
strategy:
max-parallel: 7
max-parallel: 5
fail-fast: false
matrix:
python-version: [3.8, 3.9, "3.10"]
pytorch-version:
[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]
[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]
exclude:
- pytorch-version: 1.5.1
python-version: 3.9
- pytorch-version: 1.5.1
python-version: "3.10"

- pytorch-version: 1.6.0
python-version: 3.9
- pytorch-version: 1.6.0
python-version: "3.10"

# disabling python 3.9 support with PyTorch 1.7.1 and 1.8.1, to stop repeated pytorch-version test fail.
# https://github.com/pytorch/ignite/issues/2383
- pytorch-version: 1.7.1
python-version: 3.9
- pytorch-version: 1.7.1
python-version: "3.10"

- pytorch-version: 1.8.1
python-version: 3.9
- pytorch-version: 1.8.1
Expand Down
2 changes: 1 addition & 1 deletion docker/docker.cfg
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
[DEFAULT]
build_docker_image_pytorch_version = 2.1.0-cuda12.1-cudnn8
build_docker_image_pytorch_version = 2.2.0-cuda12.1-cudnn8
build_docker_image_hvd_version = v0.28.1
build_docker_image_msdp_version = v0.8.1

0 comments on commit cf26025

Please sign in to comment.