Ubuntu下的CUDA与cuDNN安装与卸载全攻略

在深度学习和计算机视觉领域，CUDA和cuDNN是两个不可或缺的工具。CUDA（Compute Unified Device Architecture）是NVIDIA推出的并行计算平台和API模型，它使得开发者可以使用NVIDIA的GPU进行高性能计算。而cuDNN（CUDA Deep Neural Network library）则是一个针对深度神经网络的GPU加速库，它为常用的深度学习操作提供了高效的实现。

一、Ubuntu下CUDA的安装

1. 下载CUDA：首先，前往NVIDIA官网下载适合您GPU的CUDA版本。请确保选择与您的GPU型号和Ubuntu版本兼容的版本。

2. 安装CUDA：下载完成后，打开终端，转到CUDA安装包所在的目录，并运行安装命令。一般来说，这个命令是sudo sh cuda_installation_file.run，其中cuda_installation_file.run是您下载的CUDA安装包。

3. 环境配置：安装完成后，需要配置环境变量。打开~/.bashrc文件，在文件末尾添加以下信息：

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
export CUDA_HOME=/usr/local/cuda/bin
export PATH=$PATH:$LD_LIBRARY_PATH:$CUDA_HOME

然后，运行source ~/.bashrc使配置生效。

1. 验证CUDA安装：关闭当前命令行窗口，重新打开一个新的命令行窗口，运行nvcc -V命令。如果显示CUDA版本信息，则说明安装成功。

二、Ubuntu下cuDNN的安装

cuDNN通常与CUDA一起安装，因为cuDNN依赖于CUDA。如果您已经安装了CUDA，那么cuDNN应该已经自动安装好了。如果没有，您可以手动从NVIDIA官网下载并安装cuDNN。

1. 下载cuDNN：前往NVIDIA官网下载与您的CUDA版本兼容的cuDNN版本。

2. 解压cuDNN：下载完成后，解压cuDNN安装包，并将解压后的文件复制到CUDA的安装目录中。一般来说，这个目录是/usr/local/cuda/include和/usr/local/cuda/lib64。

三、Ubuntu下CUDA和cuDNN的卸载

如果您需要卸载CUDA和cuDNN，可以按照以下步骤进行：

1. 卸载CUDA：首先，运行CUDA的卸载脚本。这个脚本通常位于CUDA安装目录的bin子目录下，名为uninstall_cuda.pl或类似的名称。在终端中运行sudo ./uninstall_cuda.pl命令即可开始卸载CUDA。

2. 删除残留文件：卸载脚本运行完成后，还需要手动删除一些残留文件。运行以下命令删除CUDA相关的文件和目录：

sudo rm -rf /usr/local/cuda*

这将删除CUDA的所有文件和目录。

1. 卸载cuDNN：由于cuDNN依赖于CUDA，因此当您卸载CUDA时，cuDNN也会被自动卸载。如果您只想卸载cuDNN而不卸载CUDA，那么您需要手动删除cuDNN相关的文件。这些文件通常位于CUDA的安装目录中。运行以下命令删除cuDNN相关的文件和目录：

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

这将删除cuDNN的所有文件和目录。

请注意，在卸载CUDA和cuDNN之前，请确保您已经备份了所有重要的数据和配置文件。卸载这些工具可能会导致一些依赖它们的应用程序无法正常运行。此外，如果您在卸载后重新安装CUDA和cuDNN，可能需要重新配置环境变量。