Limit the Workaround Scope

pytorch · Sep 11, 2024 · a9b4852 · a9b4852
1 parent b3badc0
commit a9b4852
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh
@@ -15,9 +15,14 @@ else
  conda install -y conda=23.11.0
  fi
  # Please note ffmpeg is required for torchaudio, see https://github.com/pytorch/pytorch/issues/96159
- conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} ffmpeg
+ conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
  conda activate ${ENV_NAME}
- pip3 install numpy --force-reinstall
+
+ # Remove when https://github.com/pytorch/builder/issues/1985 is fixed
+ if [[ ${MATRIX_GPU_ARCH_TYPE} == 'cuda' && ${TARGET_OS} == 'linux-aarch64' ]]; then 
+ pip3 install numpy --force-reinstall
+ fi
+
  INSTALLATION=${MATRIX_INSTALLATION/"conda install"/"conda install -y"}
  TEST_SUFFIX=""