diff --git a/hpvm/set_paths.sh b/hpvm/set_paths.sh
index cbbada8e291f081114de29a860842d242983ccb2..42d1be52949ebf780d9fd7836d0429aa970472a5 100644
--- a/hpvm/set_paths.sh
+++ b/hpvm/set_paths.sh
@@ -1,14 +1,14 @@
 #!/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/
+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
-module load $CUDA_TOOLKIT_PATH
+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