diff --git a/hpvm/set_paths.sh b/hpvm/set_paths.sh index 42d1be52949ebf780d9fd7836d0429aa970472a5..8c80a2a30b68ebd301f20d08fb795ade63582f08 100644 --- a/hpvm/set_paths.sh +++ b/hpvm/set_paths.sh @@ -1,17 +1,20 @@ #!/bin/bash # These paths can be modified by the HPVM user -CUDA_TOOLKIT_PATH=/software/cuda-9.1/ -CUDA_INCLUDE_PATH=$CUDA_TOOLKIT_PATH/include -CUDA_LIB_PATH=$CUDA_TOOLKIT_PATH/lib64/ +CUDA_TOOLKIT_PATH= # Set this to the root of your CUDA Installation +if [ -z "$CUDA_TOOLKIT_PATH" ] +then + echo "ERROR: SET CUDA_TOOLKIT_PATH to the Root of your CUDA Installation" +else + CUDA_INCLUDE_PATH=$CUDA_TOOLKIT_PATH/include + CUDA_LIB_PATH=$CUDA_TOOLKIT_PATH/lib64/ + echo "Setting environment paths..." -echo "Setting environment paths..." - -# Setting CUDA paths here -export CUDA_BIN_PATH=$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!" + # Setting CUDA paths here + export CUDA_BIN_PATH=$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!" +fi