美文网首页
shell脚本切换cuda版本

shell脚本切换cuda版本

作者: wangdsh | 来源:发表于2018-08-12 15:02 被阅读0次

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!"

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

相关文章

网友评论

      本文标题:shell脚本切换cuda版本

      本文链接:https://www.haomeiwen.com/subject/fkpsbftx.html