Merge pull request #121 from coreweave/es/torch-2.10.0

feat(torch): Update PyTorch to v2.10.0

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.9.1
-    vision: 0.24.1
-    audio: 2.9.1
+  - torch: 2.10.0
+    vision: 0.25.0
+    audio: 2.10.0

.github/configurations/torch-nccl.yml

@@ -2,8 +2,8 @@ cuda: [ 12.9.1, 12.8.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.9.1
-    vision: 0.24.1
-    audio: 2.9.1
+  - torch: 2.10.0
+    vision: 0.25.0
+    audio: 2.10.0
     nccl: 2.29.2-1
-    nccl-tests-hash: '2ca8240'
+    nccl-tests-hash: 'd73ec07'

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