ubuntu安装多个cuda版本+版本切换

以下为gpt-4o生成内容:

安装多个 CUDA 版本

  1. 下载 CUDA 版本

    访问 NVIDIA CUDA Toolkit Archive 并下载你需要的 CUDA 版本的 .run 文件。

  2. 安装 CUDA

    以下是安装 CUDA 11.6 和 CUDA 12.1 的示例。调整版本号以适应你的需求。

    # 安装 CUDA 11.6
    sudo sh cuda_11.6.2_460.32.03_linux.run
    
    # 安装 CUDA 12.1
    sudo sh cuda_12.1.0_470.57.02_linux.run
    

    在安装过程中,你可以选择不安装驱动程序(如果你已经有一个兼容的驱动程序),以避免驱动程序版本冲突。

  3. 安装完成后,你会在 /usr/local/ 目录下看到多个 CUDA 版本的文件夹,例如 /usr/local/cuda-11.6/usr/local/cuda-12.1

配置环境变量和切换 CUDA 版本

你可以通过设置环境变量来切换 CUDA 版本。可以使用 update-alternatives 工具来简化这个过程。

  1. 配置 update-alternatives

    首先,为 CUDA 创建 update-alternatives 配置。

    sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-11.6 110
    sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12.1 120
    

    这两条命令将 /usr/local/cuda 指向不同的 CUDA 版本,并分别赋予它们优先级。

  2. 切换 CUDA 版本

    使用 update-alternatives 工具进行切换。

    sudo update-alternatives --config cuda
    

    你会看到一个菜单,可以选择要使用的 CUDA 版本。

    输出示例:

    There are 2 choices for the alternative cuda (providing /usr/local/cuda).
    
      Selection    Path                 Priority   Status
    ------------------------------------------------------------
    * 0            /usr/local/cuda-12.1  120       auto mode
      1            /usr/local/cuda-11.6  110       manual mode
      2            /usr/local/cuda-12.1  120       manual mode
    
    Press <enter> to keep the current choice[*], or type selection number:
    
  3. 更新环境变量

    你需要确保环境变量 PATHLD_LIBRARY_PATH 正确指向所选的 CUDA 版本。在你的 shell 配置文件(例如 .bashrc.zshrc)中添加以下内容:

    export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    

    每次切换 CUDA 版本后,记得重新加载 shell 配置文件:

    source ~/.bashrc  # 或者 source ~/.zshrc
    

示例操作

  1. 安装 CUDA 版本

    sudo sh cuda_11.6.2_460.32.03_linux.run
    sudo sh cuda_12.1.0_470.57.02_linux.run
    
  2. 配置 update-alternatives

    sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-11.6 110
    sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12.1 120
    
  3. 切换 CUDA 版本

    sudo update-alternatives --config cuda
    
  4. 更新环境变量

    .bashrc.zshrc 中添加:

    export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    

    并重新加载配置:

    source ~/.bashrc  # 或者 source ~/.zshrc
    

通过这些步骤,你可以在 Ubuntu 上安装和切换多个 CUDA 版本,以满足不同项目的需求。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值