Ubuntu下CUDA8.0和CUDA9.0共存怎么实现

在深度学习和高性能计算领域，CUDA（Compute Unified Device Architecture）是NVIDIA推出的并行计算平台和编程模型。它允许开发者使用NVIDIA GPU进行通用计算，从而加速计算密集型任务。然而，不同的深度学习框架或应用程序可能需要不同版本的CUDA库。因此，在某些情况下，用户可能需要在同一台机器上安装多个CUDA版本，并在它们之间进行切换。

本文将详细介绍如何在Ubuntu系统上实现CUDA 8.0和CUDA 9.0的共存，并展示如何在不同版本之间进行切换。

1. 准备工作

在开始之前，请确保你已经安装了NVIDIA驱动程序和CUDA工具包。如果你还没有安装，可以参考NVIDIA官方文档进行安装。

1.1 检查NVIDIA驱动

首先，检查你的系统是否已经安装了NVIDIA驱动程序：

nvidia-smi

如果命令输出显示了GPU的信息，说明NVIDIA驱动已经安装成功。

1.2 下载CUDA安装包

从NVIDIA官方网站下载CUDA 8.0和CUDA 9.0的安装包。你可以选择下载.run文件，这样可以更灵活地管理多个CUDA版本。

• CUDA 8.0下载页面
• CUDA 9.0下载页面

2. 安装CUDA 8.0和CUDA 9.0

2.1 安装CUDA 8.0

1. 首先，给下载的CUDA 8.0安装包赋予执行权限：

   chmod +x cuda_8.0.61_375.26_linux.run

1. 运行安装程序：

   sudo ./cuda_8.0.61_375.26_linux.run

1. 在安装过程中，系统会提示你选择安装选项。确保不要安装NVIDIA驱动程序（因为我们已经安装了最新的驱动程序），并且不要安装CUDA示例（除非你需要它们）。你可以选择安装CUDA工具包和CUDA示例。

2. 安装完成后，CUDA 8.0将被安装到/usr/local/cuda-8.0目录下。

2.2 安装CUDA 9.0

1. 同样，给下载的CUDA 9.0安装包赋予执行权限：

   chmod +x cuda_9.0.176_384.81_linux.run

1. 运行安装程序：

   sudo ./cuda_9.0.176_384.81_linux.run

1. 在安装过程中，同样确保不要安装NVIDIA驱动程序，并且不要安装CUDA示例。

2. 安装完成后，CUDA 9.0将被安装到/usr/local/cuda-9.0目录下。

3. 配置环境变量

为了实现CUDA 8.0和CUDA 9.0的共存，我们需要通过环境变量来切换不同的CUDA版本。

3.1 编辑.bashrc文件

打开你的.bashrc文件：

nano ~/.bashrc

在文件末尾添加以下内容：

# CUDA 8.0
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# CUDA 9.0
# export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
# export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3.2 切换CUDA版本

默认情况下，我们使用CUDA 8.0。如果你需要切换到CUDA 9.0，只需将.bashrc文件中的CUDA 8.0的路径注释掉，并取消注释CUDA 9.0的路径：

# CUDA 8.0
# export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
# export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# CUDA 9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存并关闭文件，然后运行以下命令使更改生效：

source ~/.bashrc

3.3 验证CUDA版本

你可以通过以下命令验证当前使用的CUDA版本：

nvcc --version

如果输出显示的是CUDA 8.0或CUDA 9.0，说明切换成功。

4. 使用符号链接管理CUDA版本

除了手动编辑.bashrc文件外，你还可以使用符号链接来管理CUDA版本。这种方法更加灵活，适合需要频繁切换CUDA版本的用户。

4.1 创建符号链接

首先，删除现有的/usr/local/cuda符号链接：

sudo rm -rf /usr/local/cuda

然后，创建一个指向CUDA 8.0的符号链接：

sudo ln -s /usr/local/cuda-8.0 /usr/local/cuda

或者，创建一个指向CUDA 9.0的符号链接：

sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda

4.2 验证符号链接

你可以通过以下命令验证符号链接是否指向正确的CUDA版本：

ls -l /usr/local/cuda

如果输出显示符号链接指向/usr/local/cuda-8.0或/usr/local/cuda-9.0，说明符号链接创建成功。

5. 总结

通过以上步骤，你可以在Ubuntu系统上实现CUDA 8.0和CUDA 9.0的共存，并且可以根据需要轻松切换不同的CUDA版本。无论是通过环境变量还是符号链接，你都可以灵活地管理多个CUDA版本，以满足不同应用程序的需求。

希望本文对你有所帮助！如果你有任何问题或建议，欢迎在评论区留言。