以下为gpt-4o生成内容:
安装多个 CUDA 版本
-
下载 CUDA 版本:
访问 NVIDIA CUDA Toolkit Archive 并下载你需要的 CUDA 版本的
.run
文件。 -
安装 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
在安装过程中,你可以选择不安装驱动程序(如果你已经有一个兼容的驱动程序),以避免驱动程序版本冲突。
-
安装完成后,你会在
/usr/local/
目录下看到多个 CUDA 版本的文件夹,例如/usr/local/cuda-11.6
和/usr/local/cuda-12.1
。
配置环境变量和切换 CUDA 版本
你可以通过设置环境变量来切换 CUDA 版本。可以使用 update-alternatives
工具来简化这个过程。
-
配置
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 版本,并分别赋予它们优先级。 -
切换 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:
-
更新环境变量:
你需要确保环境变量
PATH
和LD_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
示例操作
-
安装 CUDA 版本:
sudo sh cuda_11.6.2_460.32.03_linux.run sudo sh cuda_12.1.0_470.57.02_linux.run
-
配置
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
-
切换 CUDA 版本:
sudo update-alternatives --config cuda
-
更新环境变量:
在
.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 版本,以满足不同项目的需求。