Skip to content
Snippets Groups Projects
Commit b450ed7e authored by Hashim Sharif's avatar Hashim Sharif Committed by Yifan Zhao
Browse files

Adding check in set_path.sh throwing error if CUDA path not set

parent 0bc1df24
No related branches found
No related tags found
No related merge requests found
#!/bin/bash #!/bin/bash
# These paths can be modified by the HPVM user # These paths can be modified by the HPVM user
CUDA_TOOLKIT_PATH=/software/cuda-9.1/ CUDA_TOOLKIT_PATH= # Set this to the root of your CUDA Installation
CUDA_INCLUDE_PATH=$CUDA_TOOLKIT_PATH/include if [ -z "$CUDA_TOOLKIT_PATH" ]
CUDA_LIB_PATH=$CUDA_TOOLKIT_PATH/lib64/ 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
# Setting CUDA paths here export CUDA_INCLUDE_PATH=$CUDA_INCLUDE_PATH
export CUDA_BIN_PATH=$CUDA_TOOLKIT_PATH export CUDNN_PATH=$CUDA_LIB_PATH
export CUDA_INCLUDE_PATH=$CUDA_INCLUDE_PATH export LIBRARY_PATH=$CUDA_LIB_PATH:$LIBRARY_PATH
export CUDNN_PATH=$CUDA_LIB_PATH export LD_LIBRARY_PATH=$CUDA_LIB_PATH:$LD_LIBRARY_PATH
export LIBRARY_PATH=$CUDA_LIB_PATH:$LIBRARY_PATH echo "Finished setting environment paths!"
export LD_LIBRARY_PATH=$CUDA_LIB_PATH:$LD_LIBRARY_PATH fi
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