shell脚本切换cuda版本

ubuntu16.04 LTS 上有cuda8.0和cuda9.0两个版本，因为需要切换使用，所以写了一个shell脚本方便快速切换
.bashrc中cuda环境变量：

export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH"

shell脚本：

#! /bin/bash

if [ $# -ne 1 ] ; then
echo "请输入一个版本参数：8 或 9"
exit
fi

echo "版本参数：$1";
sed -ie "s/cuda-..0/cuda-$1.0/g" ~/.bashrc &&
source ~/.bashrc &&
sudo ln -snf /usr/local/cuda-${1}.0 /usr/local/cuda &&
sudo ldconfig

echo "更改完毕，请重新登录shell!"

如果本文对您有所帮助，请点击下方的“喜欢”按钮 -:)