Skip to content

Commit 20a228e

Browse files
committed
feat(torch): Update PyTorch to v2.10.0 and CUDA to 13.1.1
1 parent 4ae9dbb commit 20a228e

File tree

3 files changed

+12
-12
lines changed

3 files changed

+12
-12
lines changed
Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
cuda: [ 13.1.0 ]
1+
cuda: [ 13.1.1 ]
22
os: [ ubuntu22.04 ]
33
abi: [ 1 ]
44
include:
5-
- torch: 2.9.1
6-
vision: 0.24.1
7-
audio: 2.9.1
5+
- torch: 2.10.0
6+
vision: 0.25.0
7+
audio: 2.10.0
Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
cuda: [ 13.1.0 ]
1+
cuda: [ 13.1.1 ]
22
os: [ ubuntu22.04 ]
33
abi: [ 1 ]
44
include:
5-
- torch: 2.9.1
6-
vision: 0.24.1
7-
audio: 2.9.1
5+
- torch: 2.10.0
6+
vision: 0.25.0
7+
audio: 2.5.0
88
nccl: 2.29.2-1
9-
nccl-tests-hash: '4e5aaa7'
9+
nccl-tests-hash: 'd73ec07'

torch/Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
ARG BUILDER_BASE_IMAGE="nvidia/cuda:12.9.1-devel-ubuntu22.04"
33
ARG FINAL_BASE_IMAGE="nvidia/cuda:12.9.1-base-ubuntu22.04"
44

5-
ARG BUILD_TORCH_VERSION="2.9.1"
6-
ARG BUILD_TORCH_VISION_VERSION="0.24.1"
7-
ARG BUILD_TORCH_AUDIO_VERSION="2.9.1"
5+
ARG BUILD_TORCH_VERSION="2.10.0"
6+
ARG BUILD_TORCH_VISION_VERSION="0.25.0"
7+
ARG BUILD_TORCH_AUDIO_VERSION="2.10.0"
88
ARG BUILD_TRANSFORMERENGINE_VERSION="b10f436aa28ec8d885eb0d9bf134c320ffb6353d"
99
ARG BUILD_FLASH_ATTN_VERSION="2.7.4.post1"
1010
ARG BUILD_FLASH_ATTN_3_VERSION="1ceaa984b2f348caea18b39a98458d33b4ea7a09"

0 commit comments

Comments
 (0)