Update generate_binary_build_matrix.py

pytorch · Nov 26, 2024 · 04a487f · 04a487f
1 parent 93a3888
commit 04a487f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -148,9 +148,9 @@ def initialize_globals(channel: str, build_python_only: bool) -> None:
     else:
         PYTHON_ARCHES = PYTHON_ARCHES_DICT[channel]
     WHEEL_CONTAINER_IMAGES = {
-       "11.8": f"pytorch/manylinux-builder:cuda11.8-{DEFAULT_TAG}",
-       "12.4": f"pytorch/manylinux-builder:cuda12.4-{DEFAULT_TAG}",
-       "12.6": f"pytorch/manylinux2_28-builder:cuda12.6-{DEFAULT_TAG}",
+       "11.8": f"pytorch/manylinux-builder:cuda11.8",
+       "12.4": f"pytorch/manylinux-builder:cuda12.4",
+       "12.6": f"pytorch/manylinux2_28-builder:cuda12.6",
         **{
             gpu_arch: f"pytorch/manylinux-builder:rocm{gpu_arch}"
             for gpu_arch in ROCM_ARCHES