环境变量设置

• TVM相关

# bazel path
export PATH=/data01/user/bin:$PATH
# python venv
source /data01/user/tvm.venv/bin/activate
# enable tvm environment
export TVM_HOME=/data01/user/workspace/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/topi/python:${PYTHONPATH}
# enable cuda environment
export PATH=/data01/user/workspace/cuda/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/data01/user/workspace/cuda/lib64:/data01/user/workspace/cuda/extras/CUPTI/lib64

• CUDA 相关

export PATH=/data01/user/workspace/cuda/bin:$PATH
export LD_LIBRARY_PATH="/data01/user/workspace/cuda/lib64:/data01/user/workspace/cuda/extras/CUPTI/lib64"
CUDA_PATH=/data01/user/workspace/cuda
export CUDA_TOOLKIT_ROOT_DIR=$CUDA_PATH
export CUDA_CUDA_LIBRARY=$CUDA_PATH/lib64/stubs/
export CUDA_CUDART_LIBRARY=$CUDA_PATH/lib64
export CUDA_NVRTC_LIBRARY=$CUDA_PATH/lib64
export CUDA_CUDNN_LIBRARY=$CUDA_PATH/lib64
export CUDA_CUBLAS_LIBRARY=$CUDA_PATH/lib64

• 编译器 相关

export PATH=/data01/user/gcc-7.3.0/build/install/bin:$PATH
export LIBRARY_PATH=/data01/user/gcc-7.3.0/build/install/lib64:$LIBRARY_PATH