Update generate_binary_build_matrix.py

2025-11-02 14:34:54 +08:00 · 2024-08-22 15:52:52 -06:00
1 changed files with 1 additions and 1 deletions
--- a/.github/scripts/generate_binary_build_matrix.py
+++ b/.github/scripts/generate_binary_build_matrix.py
@ -166,7 +166,7 @@ WHEEL_CONTAINER_IMAGES = {
    "cpu-cxx11-abi": f"pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi-{DEFAULT_TAG}",
    "cpu-aarch64": f"pytorch/manylinuxaarch64-builder:cpu-aarch64-{DEFAULT_TAG}",
    "cpu-s390x": f"pytorch/manylinuxs390x-builder:cpu-s390x-{DEFAULT_TAG}",
-    "cuda-aarch64": f"pytorch/manylinuxaarch64-builder:cuda12.4-{DEFAULT_TAG}",
+    "cuda-aarch64": "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/manylinuxaarch64-builder-cuda12.4:b13abe3d3a9ddc2a055931af2d22c2ed9dab8209",
 }

 CONDA_CONTAINER_IMAGES = {