Skip to content

Commit 8e5151c

Browse files
authored
Add aarch64 conda env lib to LD_LIBRARY_PATH (#1628)
After the change on #1586, nightly aarch64 wheel fails to find `libopenblas.so` which is now installed under `/opt/conda/envs/aarch64_env/lib/` instead of the base conda `/opt/conda/lib`. Using CPU nightly wheels on aarch64 from Nov 16 then ends up with the error as described in pytorch/pytorch#114862: `Calling torch.geqrf on a CPU tensor requires compiling PyTorch with LAPACK. Please use PyTorch built with LAPACK support`. The error can be found on night build log https://github.com/pytorch/pytorch/actions/runs/6887666324/job/18735230109#step:15:4933 Fixes pytorch/pytorch#114862 I double check `2.1.[0-1]` and the current RC for 2.1.2, the issue is not there because #1586 only change builder main, thus impacting nightly. ### Testing Build nightly wheel manually on aarch64 runner and confirm that openblas is detected correctly: ``` -- Found a library with BLAS API (open). Full path: (/opt/conda/envs/aarch64_env/lib/libopenblas.so) ... -- USE_BLAS : 1 -- BLAS : open -- BLAS_HAS_SBGEMM : -- USE_LAPACK : 1 -- LAPACK : open ... ```
1 parent fc773dd commit 8e5151c

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

aarch64_linux/aarch64_ci_setup.sh

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,9 @@ set -eux -o pipefail
66

77
CONDA_PYTHON_EXE=/opt/conda/bin/python
88
CONDA_EXE=/opt/conda/bin/conda
9+
CONDA_ENV_NAME=aarch64_env
910
PATH=/opt/conda/bin:$PATH
10-
LD_LIBRARY_PATH=/opt/conda/lib:$LD_LIBRARY_PATH
11+
LD_LIBRARY_PATH=/opt/conda/envs/${CONDA_ENV_NAME}/lib/:/opt/conda/lib:$LD_LIBRARY_PATH
1112

1213
###############################################################################
1314
# Install conda
@@ -21,8 +22,8 @@ chmod +x /mambaforge.sh
2122
rm /mambaforge.sh
2223
source /opt/conda/etc/profile.d/conda.sh
2324
conda config --set ssl_verify False
24-
conda create -y -c conda-forge -n aarch64_env python=${DESIRED_PYTHON}
25-
conda activate aarch64_env
25+
conda create -y -c conda-forge -n "${CONDA_ENV_NAME}" python=${DESIRED_PYTHON}
26+
conda activate "${CONDA_ENV_NAME}"
2627

2728
if [[ "$DESIRED_PYTHON" == "3.8" ]]; then
2829
NUMPY_VERSION="1.24.4"

0 commit comments

Comments
 (0)