diff --git a/.github/configurations/torch-base.yml b/.github/configurations/torch-base.yml index 4541138..67adbe2 100644 --- a/.github/configurations/torch-base.yml +++ b/.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 diff --git a/.github/configurations/torch-nccl.yml b/.github/configurations/torch-nccl.yml index 2366f75..d572363 100644 --- a/.github/configurations/torch-nccl.yml +++ b/.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' diff --git a/torch/Dockerfile b/torch/Dockerfile index b6ab687..4207bdf 100755 --- a/torch/Dockerfile +++ b/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"