Linux 多版本 CUDA 安装

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


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 命令不会跟随软链接改变。