set_paths.sh

#!/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/

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!"