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!"
如果本文对您有所帮助,请点击下方的“喜欢”按钮 -:)