From 00433d145056aaa2c222a2c2d366103877dd482c Mon Sep 17 00:00:00 2001
From: David Corvoysier <david@huggingface.co>
Date: Mon, 3 Mar 2025 16:29:21 +0100
Subject: [PATCH] ci: use pytorch 2.6

---
 .github/workflows/linux-cuda-tests.yml | 4 ++--
 .github/workflows/linux-examples.yml   | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/linux-cuda-tests.yml b/.github/workflows/linux-cuda-tests.yml
index 84969885..e8243a33 100644
--- 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
index ee1950ad..22fda3cc 100644
--- 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: