Merge pull request #116 from coreweave/es/torch-2.9.1

feat(torch): Build with Pytorch v2.9.1 & NCCL 2.29.2-1

Author: Eta0

.github/configurations/torch-base.yml

@@ -2,6 +2,6 @@ cuda: [ 12.9.1, 12.8.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.8.0
-    vision: 0.23.0
-    audio: 2.8.0
+  - torch: 2.9.1
+    vision: 0.24.1
+    audio: 2.9.1

.github/configurations/torch-nccl.yml

@@ -2,8 +2,8 @@ cuda: [ 12.9.1, 12.8.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.8.0
-    vision: 0.23.0
-    audio: 2.8.0
-    nccl: 2.28.3-1
-    nccl-tests-hash: '528fb68'
+  - torch: 2.9.1
+    vision: 0.24.1
+    audio: 2.9.1
+    nccl: 2.29.2-1
+    nccl-tests-hash: '2ca8240'

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.8.0"
-ARG BUILD_TORCH_VISION_VERSION="0.23.0"
-ARG BUILD_TORCH_AUDIO_VERSION="2.8.0"
+ARG BUILD_TORCH_VERSION="2.9.1"
+ARG BUILD_TORCH_VISION_VERSION="0.24.1"
+ARG BUILD_TORCH_AUDIO_VERSION="2.9.1"
 ARG BUILD_TRANSFORMERENGINE_VERSION="2.4"
 ARG BUILD_FLASH_ATTN_VERSION="2.7.4.post1"
 ARG BUILD_FLASH_ATTN_3_VERSION="b36ad4ef767d2d5536ff8af2e3f720ae4eba731c"