diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index aa852a260c..fdbd187283 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -84,6 +84,14 @@ # and release builds and their validation are unchanged. CUDA_ARCHES_NO_GETTING_STARTED = ["13.4"] +# CUDA versions with no official Docker image, mirroring the omissions from +# CUDA_ARCHES_FULL_VERSION in pytorch/pytorch, which drives the docker build +# matrix. Only affects the docker matrix; wheel and libtorch builds and their +# validation are unchanged. 13.4 is here because cuda-toolkit-13-4 ships only +# in NVIDIA's preview channel (packages.nvidia.com), not in the CUDA apt repo +# that pytorch/pytorch's release Dockerfile installs the devel toolkit from. +CUDA_ARCHES_NO_DOCKER = ["13.4"] + PACKAGE_TYPES = ["wheel", "libtorch"] CXX11_ABI = "cxx11-abi" RELEASE = "release" diff --git a/tools/scripts/generate_docker_release_matrix.py b/tools/scripts/generate_docker_release_matrix.py index c8605a11ce..b0b3ab9bc7 100644 --- a/tools/scripts/generate_docker_release_matrix.py +++ b/tools/scripts/generate_docker_release_matrix.py @@ -44,7 +44,12 @@ def generate_docker_matrix( else: docker_image_version = f"{prefix}-nightly:{generate_binary_build_matrix.CURRENT_NIGHTLY_VERSION}.dev{datetime.today().strftime('%Y%m%d')}" - for cuda in generate_binary_build_matrix.CUDA_ARCHES_DICT[channel]: + cuda_arches = [ + cuda + for cuda in generate_binary_build_matrix.CUDA_ARCHES_DICT[channel] + if cuda not in generate_binary_build_matrix.CUDA_ARCHES_NO_DOCKER + ] + for cuda in cuda_arches: version = generate_binary_build_matrix.CUDA_CUDNN_VERSIONS[cuda] for image in DOCKER_IMAGE_TYPES: ret.append(