Linux系统中切换CUDA版本的完整指南（含vim使用方法）
在深度学习和高性能计算领域，经常需要在不同的CUDA版本之间切换，以满足不同项目的需求。本文将详细介绍如何在Linux系统中通过软链接切换CUDA版本的方法，并介绍了vim编辑器的使用方法。
文章目录
- Linux系统中切换CUDA版本的完整指南（含vim使用方法）
- 一、查看当前安装的CUDA版本
- 二、通过软链接切换CUDA版本
- 1. 使用vim编辑器来打开.bashrc文件
- 2. 导航到文件末尾
- 3. 进入插入模式
- 4. 添加CUDA环境变量配置
- 5. 保存并退出
- 三、使`~/.bashrc`当中的环境配置生效
- 四、验证CUDA版本切换
一、查看当前安装的CUDA版本
首先，我们需要了解系统中已安装的CUDA版本：
cd /usr/local/
ls -l | grep cuda
这将显示类似以下内容：
lrwxrwxrwx 1 root root 20 Jan 15 10:20 cuda -> /usr/local/cuda-10.0
drwxr-xr-x 1 root root 4096 Dec 10 09:15 cuda-10.0
drwxr-xr-x 1 root root 4096 Dec 10 09:15 cuda-10.1
drwxr-xr-x 1 root root 4096 Dec 10 09:15 cuda-11.0
二、通过软链接切换CUDA版本
如果只是临时切换的话，可以在当前终端中输入下列命令（仅在当前终端会话有效）：
export CUDA_HOME="/usr/local/cuda-10.1"
export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda-10.1/bin:$PATH"
如果想要永久切换的话，需要修改.bashrc文件。步骤如下：
vim是Linux系统中强大的文本编辑器，但对初学者可能有些难以上手。以下是使用vim编辑.bashrc文件的详细步骤：
1. 使用vim编辑器来打开.bashrc文件
(sudo) vim ~/.bashrc
（sudo表示使用管理员权限运行，可选）
vim有三种基本模式：
- 普通模式：默认模式，用于导航和执行命令
- 插入模式：用于编辑文本
- 命令行模式：用于执行保存、退出等操作
2. 导航到文件末尾
在普通模式下，按G键可以直接跳转到文件末尾。
3. 进入插入模式
按i键进入插入模式（屏幕左下角会显示"-- INSERT --"）。
4. 添加CUDA环境变量配置
在文件末尾添加以下内容：
# CUDA Environment Variables
export CUDA_HOME="/usr/local/cuda"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda/bin:$PATH"
请注意：上述路径中的cuda可以根据实际需要使用的CUDA版本进行修改，具体内容可以通过ls /usr/local命令来查看当前系统中安装的CUDA版本（如图所示，已经安装的CUDA版本会被显示在~/usr/local路径下）：
5. 保存并退出
- 按
Esc键退出插入模式，回到普通模式 - 输入
:wq进入命令行模式并执行保存退出操作w表示写入（保存）q表示退出
P.S. 这里推荐一个更加好用的、具有可视化界面的编辑器：geidt ，使用命令与 vim 相同：
(sudo) gedit ~/.bashrc
gedit这个文本编辑器可以避免使用上述步骤中复杂的指令调整光标位置，在可视化界面上点击即可。
Geidt编辑器的安装方式可以参考我在这篇文章中介绍的内容：Linux系统下安装Gedit文本编辑器的完整指南 。
三、使~/.bashrc当中的环境配置生效
source ~/.bashrc
四、验证CUDA版本切换
可以通过以下命令验证当前系统检测并使用的CUDA版本：
nvcc -V
希望本文对你在Linux系统中切换CUDA版本有所帮助！如有任何问题，欢迎在评论区留言交流。
