feat(torch): Build with PyTorch v2.6.0

coreweave · Jan 30, 2025 · 6bc6fb6 · 6bc6fb6
1 parent 440d844
commit 6bc6fb6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/configurations/torch-base.yml b/.github/configurations/torch-base.yml
@@ -1,6 +1,6 @@
 cuda: [ 12.8.0, 12.6.3, 12.4.1 ]
 os: [ ubuntu22.04 ]
 include:
-  - torch: 2.5.1
-    vision: 0.20.0
-    audio: 2.5.0
+  - torch: 2.6.0
+    vision: 0.21.0
+    audio: 2.6.0
diff --git a/.github/configurations/torch-nccl.yml b/.github/configurations/torch-nccl.yml
@@ -1,8 +1,8 @@
 cuda: [ 12.8.0, 12.6.3, 12.4.1 ]
 os: [ ubuntu22.04 ]
 include:
-  - torch: 2.5.1
-    vision: 0.20.0
-    audio: 2.5.0
+  - torch: 2.6.0
+    vision: 0.21.0
+    audio: 2.6.0
     nccl: 2.25.1-1
     nccl-tests-hash: 57fa979