diff --git a/hpvm/set_paths.sh b/hpvm/set_paths.sh new file mode 100644 index 0000000000000000000000000000000000000000..cbbada8e291f081114de29a860842d242983ccb2 --- /dev/null +++ b/hpvm/set_paths.sh @@ -0,0 +1,17 @@ +#!/bin/bash + +# These paths can be modified by the HPVM user +CUDA_TOOLKIT_PATH=cuda-toolkit/9.1 +CUDA_INCLUDE_PATH=/software/cuda-9.1/include +CUDA_LIB_PATH=/software/cuda-9.1/lib64/ + +echo "Setting environment paths..." + +# Setting CUDA paths here +module load $CUDA_TOOLKIT_PATH +export CUDA_INCLUDE_PATH=$CUDA_INCLUDE_PATH +export CUDNN_PATH=$CUDA_LIB_PATH +export LIBRARY_PATH=$CUDA_LIB_PATH:$LIBRARY_PATH +export LD_LIBRARY_PATH=$CUDA_LIB_PATH:$LD_LIBRARY_PATH + +echo "Finished setting environment paths!"