From 5230bdcd4761b57b441f860645f4f4eb41b7548e Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Tue, 3 Sep 2024 15:39:32 -0700 Subject: [PATCH] Bump Mellanox Tools (#3597) --- docker/build_matrix.yaml | 6 +++--- docker/generate_build_matrix.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/build_matrix.yaml b/docker/build_matrix.yaml index 40b3d6e85f..8bbab99b84 100644 --- a/docker/build_matrix.yaml +++ b/docker/build_matrix.yaml @@ -3,7 +3,7 @@ BASE_IMAGE: nvidia/cuda:12.4.1-cudnn-devel-ubuntu20.04 CUDA_VERSION: 12.4.1 IMAGE_NAME: torch-2-4-0-cu124 - MOFED_VERSION: latest-23.10 + MOFED_VERSION: latest-24.07 NVIDIA_REQUIRE_CUDA_OVERRIDE: '' PYTHON_VERSION: '3.11' PYTORCH_NIGHTLY_URL: '' @@ -54,7 +54,7 @@ BASE_IMAGE: nvidia/cuda:12.1.1-cudnn8-devel-ubuntu20.04 CUDA_VERSION: 12.1.1 IMAGE_NAME: torch-2-3-1-cu121 - MOFED_VERSION: latest-23.10 + MOFED_VERSION: latest-24.07 NVIDIA_REQUIRE_CUDA_OVERRIDE: cuda>=12.1 brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 @@ -125,7 +125,7 @@ BASE_IMAGE: nvidia/cuda:12.1.1-cudnn8-devel-ubuntu20.04 CUDA_VERSION: 12.1.1 IMAGE_NAME: torch-2-2-2-cu121 - MOFED_VERSION: latest-23.10 + MOFED_VERSION: latest-24.07 NVIDIA_REQUIRE_CUDA_OVERRIDE: cuda>=12.1 brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py index c908df52a4..61c3f4cdbf 100644 --- a/docker/generate_build_matrix.py +++ b/docker/generate_build_matrix.py @@ -231,7 +231,7 @@ def _main(): if not cuda_version or interconnect == 'EFA': entry['MOFED_VERSION'] = '' else: - entry['MOFED_VERSION'] = 'latest-23.10' + entry['MOFED_VERSION'] = 'latest-24.07' # Skip EFA drivers if not using EFA if interconnect != 'EFA':