ci: update with Torch released 2.6

Lightning-AI · Jan 30, 2025 · db47dfb · db47dfb
1 parent 5fc2e0b
commit db47dfb
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 10 deletions.
diff --git a/.azure/gpu-unittests.yml b/.azure/gpu-unittests.yml
@@ -29,9 +29,6 @@ jobs:
           docker-image: "ubuntu22.04-cuda11.8.0-py3.10-torch2.0"
           torch-ver: "2.0"
         "PyTorch | 2.X stable":
-          docker-image: "ubuntu22.04-cuda12.1.1-py3.11-torch2.5"
-          torch-ver: "2.5"
-        "PyTorch | 2.X future":
           docker-image: "ubuntu22.04-cuda12.4.1-py3.11-torch2.6"
           torch-ver: "2.6"
     # how long to run the job before automatically cancelling

diff --git a/.github/workflows/ci-tests.yml b/.github/workflows/ci-tests.yml
@@ -52,14 +52,14 @@ jobs:
           - { os: "macOS-13", python-version: "3.10", pytorch-version: "2.0.1" }
           # using the ARM based M1 machine
           - { os: "macOS-14", python-version: "3.10", pytorch-version: "2.0.1" }
-          - { os: "macOS-14", python-version: "3.12", pytorch-version: "2.5.0" }
+          - { os: "macOS-14", python-version: "3.12", pytorch-version: "2.6.0" }
           # some windows
           - { os: "windows-2022", python-version: "3.10", pytorch-version: "2.0.1" }
-          - { os: "windows-2022", python-version: "3.12", pytorch-version: "2.5.0" }
+          - { os: "windows-2022", python-version: "3.12", pytorch-version: "2.6.0" }
           # Future released version
-          - { os: "ubuntu-22.04", python-version: "3.11", pytorch-version: "2.6.0" }
-          - { os: "macOS-14", python-version: "3.11", pytorch-version: "2.6.0" }
-          - { os: "windows-2022", python-version: "3.11", pytorch-version: "2.6.0" }
+          - { os: "ubuntu-22.04", python-version: "3.11", pytorch-version: "2.7.0" }
+          - { os: "macOS-14", python-version: "3.11", pytorch-version: "2.7.0" }
+          - { os: "windows-2022", python-version: "3.11", pytorch-version: "2.7.0" }
     env:
       FREEZE_REQUIREMENTS: ${{ ! (github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/heads/release/')) }}
       TOKENIZERS_PARALLELISM: false
@@ -104,7 +104,7 @@ jobs:
           cache-references: true
 
       - name: Switch to PT test URL
-        if: ${{ matrix.pytorch-version == '2.6.0' }}
+        if: ${{ matrix.pytorch-version == '2.7.0' }}
         run: echo 'PIP_EXTRA_INDEX_URL=--extra-index-url https://download.pytorch.org/whl/test/cpu/' >> $GITHUB_ENV
       - name: Install pkg
         timeout-minutes: 25

diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -72,7 +72,6 @@ jobs:
           - { python: "3.11", pytorch: "2.3.1", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.11", pytorch: "2.4.1", cuda: "12.1.1", ubuntu: "22.04" }
           - { python: "3.11", pytorch: "2.5.0", cuda: "12.1.1", ubuntu: "22.04" }
-          # the future version - test or RC version
           - { python: "3.11", pytorch: "2.6.0", cuda: "12.4.1", ubuntu: "22.04" }
     steps:
       - uses: actions/checkout@v4