wip

pytorch · Jan 8, 2025 · 95cbb48 · 95cbb48
1 parent 1c3a48a
commit 95cbb48
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/.github/scripts/install_fbgemm.sh b/.github/scripts/install_fbgemm.sh
@@ -10,10 +10,8 @@ echo "CHANNEL: ${CHANNEL}"
 echo "CONDA_ENV: ${CONDA_ENV}"
 
 if [[ $CU_VERSION = cu* ]]; then
-    # conda env config vars set -p ${CONDA_ENV}  \
-    #     LD_LIBRARY_PATH="/usr/local/lib:${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}" \
-    #     CUDNN_INCLUDE_DIR="${CUDA_HOME}/include" \
-    #     CUDNN_LIBRARY="${CUDA_HOME}/lib64"
+    conda env config vars set -p ${CONDA_ENV}  \
+        LD_LIBRARY_PATH="/usr/local/lib:${CUDA_HOME}/lib64:${CONDA_ENV}/lib:${LD_LIBRARY_PATH}"
 
     echo "################################################################################"
     echo "[NOVA] Finding libnvrtc.so ..."