Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

Commit 8748ef7

Browse files
authored
Work-around aarch64 conda installed numpy 2.x version. (#1984)
* Work-around aarch64 conda installed numpy 2.x version. * Does anyone know what is fmpeg? :) * Do not hard code numpy version * Limit the Workaround Scope
1 parent bd1ddfa commit 8748ef7

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

.github/scripts/validate_binaries.sh

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,12 @@ else
1717
# Please note ffmpeg is required for torchaudio, see https://github.com/pytorch/pytorch/issues/96159
1818
conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
1919
conda activate ${ENV_NAME}
20+
21+
# Remove when https://github.com/pytorch/builder/issues/1985 is fixed
22+
if [[ ${MATRIX_GPU_ARCH_TYPE} == 'cuda' && ${TARGET_OS} == 'linux-aarch64' ]]; then
23+
pip3 install numpy --force-reinstall
24+
fi
25+
2026
INSTALLATION=${MATRIX_INSTALLATION/"conda install"/"conda install -y"}
2127
TEST_SUFFIX=""
2228

0 commit comments

Comments
 (0)