Skip to content
Snippets Groups Projects
Commit 9c4f4731 authored by Akash Kothari's avatar Akash Kothari
Browse files

Removed 'module load' and setting CUDA_TOOLKIT_ROOT_DIR

parent e59962ee
No related branches found
No related tags found
No related merge requests found
#!/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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment