在这篇文章当中，介绍了如何在机器上查看可以使用的cuda版本，同时有暂时更换cuda的方法以及永久更换cuda的方法。但是无论是哪种，都需要手动输入命令。于是我让ai写了一个脚本，能够通过运行脚本快速地更换cuda版本：

首先还是需要查看机器能够使用的cuda有哪些：

nvcc -V

然后脚本内容如下：

# Accept version as parameter, default to 111 if not provided
CUDA_VERSION=${1:-111}

# Format the version with a dot before the last digit (e.g., 113 -> 11.3)
FORMATTED_VERSION=$(echo $CUDA_VERSION | sed 's/\(.*\)\(.\)$/\1.\2/')

# Set environment variables
export CUDA_HOME=/usr/local/cuda-$FORMATTED_VERSION
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

echo "CUDA environment set to version $FORMATTED_VERSION"

复制内容并粘贴在sh文件中，运行时假如想使用11.7cuda，那么采用如下的命令：

source set_cuda.sh 118

注意source非常重要，因为如果不source则会开一个新的终端完成脚本文件中内容，而非在当前的终端