ci: use pytorch 2.6

huggingface · Mar 3, 2025 · 00433d1 · 00433d1
1 parent ae25cc0
commit 00433d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/linux-cuda-tests.yml b/.github/workflows/linux-cuda-tests.yml
@@ -22,14 +22,14 @@ jobs:
     uses: ./.github/workflows/python-quality.yml
   test-ubuntu-cuda:
     needs: [check-commits, python-quality]
-    runs-on: 
+    runs-on:
       group: aws-g5-4xlarge-plus
     strategy:
       fail-fast: false
       matrix:
         cuda-version: ["11.8", "12.1", "12.4"]
     container:
-      image: pytorch/pytorch:2.4.0-cuda${{ matrix.cuda-version }}-cudnn9-devel
+      image: pytorch/pytorch:2.6.0-cuda${{ matrix.cuda-version }}-cudnn9-devel
       options: --gpus 0
 
     steps:

diff --git a/.github/workflows/linux-examples.yml b/.github/workflows/linux-examples.yml
@@ -22,14 +22,14 @@ jobs:
     uses: ./.github/workflows/python-quality.yml
   run-examples:
     needs: [check-commits, python-quality]
-    runs-on: 
+    runs-on:
       group: aws-g5-4xlarge-plus
     strategy:
       fail-fast: false
       matrix:
         device: ["cpu", "cuda"]
     container:
-      image: pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
+      image: pytorch/pytorch:2.6.0-cuda12.1-cudnn9-devel
       options: --gpus 0
 
     steps: