终端动态切换cuda环境

最新推荐文章于 2023-12-11 09:16:36 发布

知识在于分享

最新推荐文章于 2023-12-11 09:16:36 发布

阅读量650

点赞数 2

分类专栏：工具类文章标签： shell

本文链接：https://blog.csdn.net/baidu_40840693/article/details/121222470

版权

工具类专栏收录该内容

172 篇文章 2 订阅

订阅专栏

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