File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -58,14 +58,12 @@ echo "CONDA_ENV: ${CONDA_ENV}"
58
58
# shellcheck disable=SC2155
59
59
export CONDA_PREFIX=$( conda run -n " ${CONDA_ENV} " printenv CONDA_PREFIX)
60
60
61
- find / -name * cuda*
62
-
63
- # Setting LD_LIBRARY_PATH fixes the runtime error with fbgemm_gpu not
61
+ # Set LD_LIBRARY_PATH to fix the runtime error with fbgemm_gpu not
64
62
# being able to locate libnvrtc.so
63
+ # NOTE: The order of the entries in LD_LIBRARY_PATH matters
65
64
echo " [NOVA] Setting LD_LIBRARY_PATH ..."
66
65
conda env config vars set -n ${CONDA_ENV} \
67
- LD_LIBRARY_PATH=" /usr/local/lib:/usr/lib64:${CONDA_PREFIX} /lib:${LD_LIBRARY_PATH} "
68
-
66
+ LD_LIBRARY_PATH=" ${CONDA_PREFIX} /lib:/usr/local/lib:/usr/lib64:${LD_LIBRARY_PATH} "
69
67
70
68
# install pytorch
71
69
# switch back to conda once torch nightly is fixed
You can’t perform that action at this time.
0 commit comments