Skip to content
Snippets Groups Projects
Commit e0a8fdb1 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Merge branch 'approx_hpvm_hotfix' into approx_hpvm_reorg

parents 0b634585 53d07733
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=/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
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