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

Add a shell script to set CUDA paths

parent 6c72a5e7
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/
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!"
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