Linux系统中切换CUDA版本的完整指南（含vim使用方法）

Linux系统中切换CUDA版本的完整指南（含vim使用方法）

在深度学习和高性能计算领域，经常需要在不同的CUDA版本之间切换，以满足不同项目的需求。本文将详细介绍如何在Linux系统中通过软链接切换CUDA版本的方法，并介绍了vim编辑器的使用方法。

一、查看当前安装的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. 保存并退出

1. 按Esc键退出插入模式，回到普通模式
2. 输入:wq进入命令行模式并执行保存退出操作
   • w表示写入（保存）
   • q表示退出

P.S. 这里推荐一个更加好用的、具有可视化界面的编辑器：geidt ，使用命令与 vim 相同：

(sudo) gedit ~/.bashrc

gedit这个文本编辑器可以避免使用上述步骤中复杂的指令调整光标位置，在可视化界面上点击即可。

Geidt编辑器的安装方式可以参考我在这篇文章中介绍的内容：Linux系统下安装Gedit文本编辑器的完整指南 。

三、使~/.bashrc当中的环境配置生效

source ~/.bashrc

四、验证CUDA版本切换

可以通过以下命令验证当前系统检测并使用的CUDA版本：

nvcc -V

希望本文对你在Linux系统中切换CUDA版本有所帮助！如有任何问题，欢迎在评论区留言交流。