feat(torch): Update PyTorch to v2.9.1

Author: Eta0

.github/configurations/torch-base.yml

@@ -2,6 +2,6 @@ cuda: [ 13.0.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.9.0
-    vision: 0.24.0
-    audio: 2.9.0
+  - torch: 2.9.1
+    vision: 0.24.1
+    audio: 2.9.1

.github/configurations/torch-nccl.yml

@@ -2,8 +2,8 @@ cuda: [ 13.0.1 ]
 os: [ ubuntu22.04 ]
 abi: [ 1 ]
 include:
-  - torch: 2.9.0
-    vision: 0.24.0
-    audio: 2.9.0
+  - torch: 2.9.1
+    vision: 0.24.1
+    audio: 2.9.1
     nccl: 2.28.7-1
     nccl-tests-hash: 'd70ae26'

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.0"
-ARG BUILD_TORCH_VISION_VERSION="0.24.0"
-ARG BUILD_TORCH_AUDIO_VERSION="2.9.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="b10f436aa28ec8d885eb0d9bf134c320ffb6353d"
 ARG BUILD_FLASH_ATTN_VERSION="2.7.4.post1"
 ARG BUILD_FLASH_ATTN_3_VERSION="1ceaa984b2f348caea18b39a98458d33b4ea7a09"