diff --git a/.github/scripts/install_fbgemm.sh b/.github/scripts/install_fbgemm.sh index c607b6e9b..5ac715dff 100644 --- 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 ..."