Ubuntu 切换 cuda 版本
查看当前 cuda 版本通过修改环境变量切换 cuda 版本通过修改软链接切换 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命令打开环境变量，使用如下命令进行修改：
键盘 a 进入编辑模式；键盘 Esc 推出编辑模式；非编辑模式下，键入 :wq （注意冒号）保存并退出；非编辑模式下，键入 :q 退出；非编辑模式下，键入 :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，如果想切换其他版本，只需再次修改软链接。
相关链接
发表评论