From 5fadff76116fa6064e7e0d99be718103516da2e9 Mon Sep 17 00:00:00 2001 From: Huanyu He Date: Wed, 22 Jan 2025 15:33:30 -0800 Subject: [PATCH] and LB_LIBRARY_PATH for cpu case in install_fbgemm.sh (#2699) Summary: # context * `LB_LIBRARY_PATH` is missing the required path for cpu case, leading to test failure at https://github.com/pytorch/torchrec/actions/runs/12914965105 ``` ImportError: /lib64/libstdc++.so.6: version `GLIBCXX_3.4.29' not found (required by /__w/_temp/conda_environment_12914965105/lib/python3.12/site-packages/tensordict/_C.so) ``` * this diff adding the missing branch when there is only cpu available Reviewed By: vmoens Differential Revision: D68524715 --- .github/scripts/install_fbgemm.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.github/scripts/install_fbgemm.sh b/.github/scripts/install_fbgemm.sh index 0b0b3e347..dc643deaa 100644 --- a/.github/scripts/install_fbgemm.sh +++ b/.github/scripts/install_fbgemm.sh @@ -15,6 +15,10 @@ if [[ $CU_VERSION = cu* ]]; then echo "[NOVA] Setting LD_LIBRARY_PATH ..." conda env config vars set -p ${CONDA_ENV} \ LD_LIBRARY_PATH="/usr/local/lib:${CUDA_HOME}/lib64:${CONDA_ENV}/lib:${LD_LIBRARY_PATH}" +else + echo "[NOVA] Setting LD_LIBRARY_PATH ..." + conda env config vars set -p ${CONDA_ENV} \ + LD_LIBRARY_PATH="/usr/local/lib:${CONDA_ENV}/lib:${LD_LIBRARY_PATH}" fi if [ "$CHANNEL" = "nightly" ]; then