Linux 多版本 CUDA 安装

发布于 2023-11-21  830 次阅读

CUDA 下载地址（历史版本）

https://developer.nvidia.com/cuda-toolkit-archive

步骤

以 cuda_11.7.1_515.65.01_linux.ru 为例

  1. 下载对应版本 .run 文件后 sudo sh cuda_11.7.1_515.65.01_linux.ru，然后会有一小段时间的卡顿，是因为在解压文件，耐心等待即可，随后会来到如下界面：

file

  1. 输入 accept 后回车，来到如下界面：

file

  1. 如果主机上没有安装过 Nvidia 驱动，则需要选中 Driver 下的驱动，我是安装过了，所以这里要取消勾选驱动，最终如下：

file

  1. 确认选择无误后，选中 Install 回车即可。
  2. 如果已经安装过 CUDA，会有如下提示：

file

  1. 这里根据需求自行选择即可。实际上后面自己也可以使用软连接命令调整。我是现在需要使用到 CUDA 11.7，所以选择 Yes。安装开始后耐心等待即可。

多版本 CUDA 切换

默认环境变量是使用 /usr/local/cuda ，而这个文件实际上是一个软链接，链接到实际的 CUDA文件上，因此你可以自行解除原来的链接，再链接到你需要的 CUDA 文件上。链接操作可以搜索 ln 命令的使用；解除链接推荐使用 unlink /usr/local/cuda。

此外，还需要在 .bashrc （也可能是 .zshrc，取决于你用的 shell）中加入：

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
export CUDA_HOME=/usr/local/cuda

注意顺序，有的文章给的是：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

这样会导致 nvcc 命令不会跟随软链接改变。