diff --git a/ML-Frameworks/pytorch-aarch64/build-torch-ao-wheel.sh b/ML-Frameworks/pytorch-aarch64/build-torch-ao-wheel.sh index 5637977b..f772c41b 100755 --- a/ML-Frameworks/pytorch-aarch64/build-torch-ao-wheel.sh +++ b/ML-Frameworks/pytorch-aarch64/build-torch-ao-wheel.sh @@ -25,7 +25,7 @@ set -eux -o pipefail PYTHON_VERSION="3.10" # Specify DOCKER_IMAGE_MIRROR if you want to use a mirror of hub.docker.com -IMAGE_NAME="${DOCKER_IMAGE_MIRROR:-}pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-a040006da76a51c4f660331e9abd3affe5a4bd81" +IMAGE_NAME="${DOCKER_IMAGE_MIRROR:-}pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-d8be0384e085f551506bd739678109fa0f5ee7ac" TORCH_BUILD_CONTAINER_ID_FILE="${PWD}/.torch_ao_build_container_id" TEST_VENV=aarch64_env_test_torch_ao