linux切换cuda脚本

Linux下切换CUDA环境通常涉及到设置CUDA路径、更新环境变量和运行特定版本的CUDA工具链。如果你需要编写一个简单的脚本来自动化这个过程，可以参考以下几个步骤：

1. **创建一个bash脚本**：
- 可能命名为`switch_cuda.sh`

   #!/bin/bash

   # 定义CUDA版本目录，这里假设你有多个版本并分别存储在/home/user/cuda-10.0, /home/user/cuda-11.0等
   CUDA_VERSIONS=("10.0" "11.0") # 更改为你实际的CUDA版本路径

   # 遍历CUDA版本并设置相应的PATH变量
   for version in "${CUDA_VERSIONS[@]}"
   do
       export PATH="/home/user/cuda-$version/bin:$PATH"
       export LD_LIBRARY_PATH="/home/user/cuda-$version/lib64:$LD_LIBRARY_PATH"
       echo "Switching to CUDA $version..."
       # 添加其他可能需要配置的环境变量，如CUDA_HOME
       export CUDA_HOME="/home/user/cuda-$version"
   done

2. **添加执行权限**：

   chmod +x switch_cuda.sh

3. **调用脚本**：
当你需要切换到特定的CUDA版本时，只需运行：

   ./switch_cuda.sh

注意：
- 确保替换上述示例中的路径为你系统上实际的CUDA安装位置。
- 如果你使用的是nvidia-smi这样的命令行工具，可能还需要更新其别名指向新版本。