cuda_switch.sh 是一个用于管理和切换系统中 CUDA 版本的 Bash 脚本工具。它支持以下功能:
显示当前 CUDA 信息:
PATH 和 LD_LIBRARY_PATH)切换 CUDA 版本:
12.6)latest)prev 或 next)帮助信息:
./cuda_switch.sh [选项]
<版本号>:切换到指定的 CUDA 版本(例如 12.6)。latest:切换到最新的 CUDA 版本。prev:切换到前一个 CUDA 版本。next:切换到后一个 CUDA 版本。help 或 -h:显示帮助信息。显示当前 CUDA 信息:
./cuda_switch.sh
切换到 CUDA 12.6:
./cuda_switch.sh 12.6
切换到最新版本:
./cuda_switch.sh latest
显示帮助信息:
./cuda_switch.sh help
~/.zshrc 文件中的 CUDA 路径,并生成临时脚本以立即应用环境变量更改。/usr/local/cuda-<版本号> 的符号链接。source ~/.zshrc 以使更改生效。脚本会自动检测系统中已安装的 CUDA 版本,无需手动配置版本列表。