and LB_LIBRARY_PATH for cpu case in install_fbgemm.sh

Summary: # context * Differential Revision: D68524715
pytorch · Jan 22, 2025 · 8eeb7d7 · 8eeb7d7
1 parent 519f193
commit 8eeb7d7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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