source envs.sh 11.1
envs.sh
#!/bin/bash
if [ ! -n "$1" ] ;then
echo "请使用正确的切换方式, 例如source envs.sh 11.1"
else
if [ $1 == "11.1" ]
then
if [ -d "/usr/local/cuda-11.1/" ];then
echo "select cuda 11.1"
export CUDA_HOME=/usr/local/cuda-11.1
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
else
echo "/usr/local/cuda-11.1/ 文件夹不存在"
fi
elif [ $1 == "9.0" ]
then
if [ -d "/usr/local/cuda-9.0/" ];then
echo "select cuda 9.0"
export CUDA_HOME=/usr/local/cuda-9.0
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
else
echo "/usr/local/cuda-9.0/ 文件夹不存在"
fi
elif [ $1 == "11.0" ]
then
if [ -d "/usr/local/cuda-11.0/" ];then
echo "select cuda 11.0"
export CUDA_HOME=/usr/local/cuda-11.0
export PATH=/usr/local/cuda-11.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH
else
echo "/usr/local/cuda-11.0/ 文件夹不存在"
fi
elif [ $1 == "10.0" ]
then
if [ -d "/usr/local/cuda-10.0/" ];then
echo "select cuda 10.0"
export CUDA_HOME=/usr/local/cuda-10.0
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
else
echo "/usr/local/cuda-10.0/ 文件夹不存在"
fi
elif [ $1 == "10.1" ]
then
if [ -d "/usr/local/cuda-10.1/" ];then
echo "select cuda 10.1"
export CUDA_HOME=/usr/local/cuda-10.1
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
else
echo "/usr/local/cuda-10.1/ 文件夹不存在"
fi
else
echo "请输入正确的环境名,参考格式11.1或者10.0等"
fi
fi
或者
#!/bin/bash
ver_lst=(
11.1
9.0
11.0
10.0
10.1
)
if [ ! -n "$1" ] ;then
echo "请使用正确的切换方式, 例如source envs.sh 11.1"
else
version=$1
if [[ "${ver_lst[@]}" =~ "${version}" ]]
then
if [ -d "/usr/local/cuda-"${version} ];then
echo "select cuda "${version}
export CUDA_HOME=/usr/local/cuda-${version}
export PATH=/usr/local/cuda-${version}/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-${version}/lib64:$LD_LIBRARY_PATH
else
echo "/usr/local/cuda-${version}/ 文件夹不存在"
fi
fi
fi