cuda切换

#写入切换函数
#*******   switch cuda
function _switch_cuda {
   v=$1
   export PATH=/usr/local/cuda-$v/bin:$PATH
   export CUDADIR=/usr/local/cuda-$v
   export CUDA_HOME=/usr/local/cuda-$v
   export LD_LIBRARY_PATH=/usr/local/cuda-$v/lib64:$LD_LIBRARY_PATH
   nvcc --version
}
_switch_cuda 8.0  #在此更改版本
#*******