logo
0
0
Login
Shiqi Wang<qtec@outlook.com>
cuda版本

CUDA 版本切换工具 (cuda_switch.sh)

功能简介

cuda_switch.sh 是一个用于管理和切换系统中 CUDA 版本的 Bash 脚本工具。它支持以下功能:

  1. 显示当前 CUDA 信息

    • 当前 CUDA 版本
    • NVIDIA 驱动版本
    • 可用的 CUDA 版本列表
    • 环境变量(如 PATHLD_LIBRARY_PATH
    • CUDA 安装路径
  2. 切换 CUDA 版本

    • 支持通过版本号直接切换(如 12.6
    • 支持切换到最新版本(latest
    • 支持切换到前一个或后一个版本(prevnext
  3. 帮助信息

    • 提供详细的用法说明和可用版本列表

使用方法

基本用法

./cuda_switch.sh [选项]

选项

  • 无参数:显示当前 CUDA 信息。
  • <版本号>:切换到指定的 CUDA 版本(例如 12.6)。
  • latest:切换到最新的 CUDA 版本。
  • prev:切换到前一个 CUDA 版本。
  • next:切换到后一个 CUDA 版本。
  • help-h:显示帮助信息。

示例

  1. 显示当前 CUDA 信息:

    ./cuda_switch.sh
  2. 切换到 CUDA 12.6:

    ./cuda_switch.sh 12.6
  3. 切换到最新版本:

    ./cuda_switch.sh latest
  4. 显示帮助信息:

    ./cuda_switch.sh help

特性

  • 自动化切换:脚本会自动更新 ~/.zshrc 文件中的 CUDA 路径,并生成临时脚本以立即应用环境变量更改。
  • 版本验证:确保目标版本在可用版本列表中,避免无效切换。
  • 用户友好:提供清晰的输出和帮助信息,便于使用。

注意事项

  • 请确保已安装所需的 CUDA 版本,并正确配置了 /usr/local/cuda-<版本号> 的符号链接。
  • 切换后,建议重新打开终端或运行 source ~/.zshrc 以使更改生效。

可用版本

脚本会自动检测系统中已安装的 CUDA 版本,无需手动配置版本列表。

About

No description, topics, or website provided.
Language
Shell89.8%
Markdown10.2%