【Ubuntu 切换 cuda 版本】

查看当前 cuda 版本

nvcc -V

不推荐使用

nvidia-smi

二者得到的 cuda 版本可能不相同，nvcc -V对应 CUDA Runtime API，而nvidia-smi对应 CUDA Driver API。一般来说，如果 CUDA Driver API 版本高于 CUDA Runtime API 版本，使用时不会有问题。

通过修改环境变量切换 cuda 版本

一般来说 cuda 安装在 /usr/local 路径下；可通过ls -l /usr/local命令查看已安装的 cuda，显示如下：

其中, /usr/local/cuda是一个软链接，可以看到，其指向 /usr/local/cuda-10.0。然而，这并不表示当前使用的 cuda 版本一定是 10.0。实际上，当前 cuda 版本取决于默认路径。

使用echo $PATH命令查看默认路径，结果如下：

第一个指向真正的 cuda 地址，因此当前 cuda 版本实际为 10.2，使用nvcc -V确认如下：

如果想切换 cuda 版本，需要更改默认路径，将需要的 cuda 版本路径加入其中。
使用vim ~/.bashrc命令打开环境变量，使用如下命令进行修改：

1. 键盘 a 进入编辑模式；
2. 键盘 Esc 推出编辑模式；
3. 非编辑模式下，键入 :wq （注意冒号）保存并退出；
4. 非编辑模式下，键入 :q 退出；
5. 非编辑模式下，键入 :e！ 将档案还原至最原始状态；

退出后使用 source ~/.bashrc 加载更新后的环境变量。

假设想修改 cuda 版本到 10.0 ，需要打开的环境变量文档最后加入下图所示语句：

保存并加载后再次使用 nvcc -V 查看：

cuda 版本成功切换到 10.0.

通过修改软链接切换 cuda 版本

修改环境变量如下：

删除当前 cuda 软链接，并建立新的软链接：

sudo rm -rf /usr/local/cuda # 删除旧软链接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda # 建立新链接 cuda -> cuda-10.0

这样 cuda 版本就切换到了 10.0，如果想切换其他版本，只需再次修改软链接。