Ubuntu 22.04 CUDA多版本安装完全指南

Ubuntu 22.04 CUDA多版本安装完全指南

快速切换CUDA 11.1和11.8教程

在深度学习和GPU计算中，CUDA安装是至关重要的步骤。对于使用Ubuntu 22.04系统的用户，有时需要管理多版本CUDA以适应不同项目需求。本教程将详细指导你如何安装CUDA 11.1和11.8，并实现CUDA切换的快速方法，即使是小白也能轻松跟随。

1. 准备工作

在开始前，请确保你的Ubuntu 22.04系统已更新，并安装了NVIDIA驱动程序。打开终端，运行以下命令：

    sudo apt updatesudo apt upgradesudo apt install nvidia-driver-525  # 建议安装最新驱动  

重启系统后，使用nvidia-smi命令验证驱动安装成功。

2. 安装CUDA 11.8

首先，访问NVIDIA官网下载CUDA 11.8的runfile安装文件。在终端中执行：

    wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.runsudo sh cuda_11.8.0_520.61.05_linux.run --toolkit --silent --override  

安装过程中，确保不安装驱动程序（因为已提前安装）。安装完成后，CUDA 11.8将默认安装在/usr/local/cuda-11.8目录。

3. 安装CUDA 11.1

类似地，下载CUDA 11.1的runfile文件并安装：

    wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.runsudo sh cuda_11.1.1_455.32.00_linux.run --toolkit --silent --override  

CUDA 11.1将安装在/usr/local/cuda-11.1目录。现在，你的系统已拥有两个CUDA版本，为多版本CUDA管理打下基础。

4. 配置多版本CUDA环境

为了实现CUDA切换，需要更新环境变量。编辑~/.bashrc文件：

    nano ~/.bashrc  

在文件末尾添加以下内容，定义别名以便快速切换：

    # CUDA版本切换配置export CUDA11_8_PATH=/usr/local/cuda-11.8export CUDA11_1_PATH=/usr/local/cuda-11.1alias switch_cuda11.8="export PATH=$CUDA11_8_PATH/bin:$PATH && export LD_LIBRARY_PATH=$CUDA11_8_PATH/lib64:$LD_LIBRARY_PATH"alias switch_cuda11.1="export PATH=$CUDA11_1_PATH/bin:$PATH && export LD_LIBRARY_PATH=$CUDA11_1_PATH/lib64:$LD_LIBRARY_PATH"# 默认使用CUDA 11.8switch_cuda11.8  

保存文件后，运行source ~/.bashrc使更改生效。

5. 快速切换CUDA版本

现在，你可以通过简单的终端命令切换CUDA版本。例如，切换到CUDA 11.1：

    switch_cuda11.1  

切换到CUDA 11.8：

    switch_cuda11.8  

上图展示了CUDA切换的流程示意图，帮助你直观理解版本管理。

6. 验证安装和切换

每次切换后，使用以下命令验证当前CUDA版本：

    nvcc --version  

输出应显示对应的CUDA版本号。此外，运行nvidia-smi确认GPU驱动正常工作。

7. 结论

通过本教程，你已在Ubuntu 22.04上成功完成了CUDA安装并配置了多版本CUDA环境。使用别名命令，可以轻松实现CUDA切换，提高工作效率。记得在实际项目中根据需求选择合适的CUDA版本。

如果你遇到问题，请参考NVIDIA官方文档或社区论坛。祝你在GPU计算中顺利前行！