Release 2.5 Use pypi for cuda 12.4 builds (#5776)

We will be releasing 12.4 builds to pypi. Hence use 12.4 rather then 12.1
pytorch · Oct 17, 2024 · 1969a06 · 1969a06
1 parent 2b66760
commit 1969a06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -310,7 +310,7 @@ def get_wheel_install_command(
         else:
             raise ValueError("Split build is not supported for this configuration. It is only supported for CUDA 11.8, 12.1, 12.4 on Linux nightly builds.")
     if  channel == RELEASE and (not use_only_dl_pytorch_org) and (
-        (gpu_arch_version == "12.1" and os == LINUX)
+        (gpu_arch_version == "12.4" and os == LINUX)
         or (
             gpu_arch_type == CPU
             and os in [WINDOWS, MACOS, MACOS_ARM64]