Merge pull request #106 from coreweave/es/torch-2.8.0

feat(torch): Build with PyTorch v2.8.0

Author: wbrown

.github/configurations/torch-base.yml

@@ -1,7 +1,7 @@
-cuda: [ 12.9.1, 12.8.1, 12.6.3 ]
+cuda: [ 12.9.1, 12.8.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.7.1
-    vision: 0.22.1
-    audio: 2.7.1
+  - torch: 2.8.0
+    vision: 0.23.0
+    audio: 2.8.0

.github/configurations/torch-nccl.yml

@@ -1,9 +1,9 @@
-cuda: [ 12.9.1, 12.8.1, 12.6.3 ]
+cuda: [ 12.9.1, 12.8.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.7.1
-    vision: 0.22.1
-    audio: 2.7.1
-    nccl: 2.27.6-1
-    nccl-tests-hash: '7c12c62'
+  - torch: 2.8.0
+    vision: 0.23.0
+    audio: 2.8.0
+    nccl: 2.27.7-1
+    nccl-tests-hash: '13b2c72'

torch/Dockerfile

@@ -2,9 +2,9 @@
 ARG BUILDER_BASE_IMAGE="nvidia/cuda:12.9.1-devel-ubuntu22.04"
 ARG FINAL_BASE_IMAGE="nvidia/cuda:12.9.1-base-ubuntu22.04"
 
-ARG BUILD_TORCH_VERSION="2.7.1"
-ARG BUILD_TORCH_VISION_VERSION="0.22.1"
-ARG BUILD_TORCH_AUDIO_VERSION="2.7.1"
+ARG BUILD_TORCH_VERSION="2.8.0"
+ARG BUILD_TORCH_VISION_VERSION="0.23.0"
+ARG BUILD_TORCH_AUDIO_VERSION="2.8.0"
 ARG BUILD_TRANSFORMERENGINE_VERSION="2.4"
 ARG BUILD_FLASH_ATTN_VERSION="2.7.4.post1"
 ARG BUILD_FLASH_ATTN_3_VERSION="b36ad4ef767d2d5536ff8af2e3f720ae4eba731c"