Ubuntu安装和卸载CUDA和CUDNN的实现

这是Ubuntu安装和卸载CUDA和CUDNN的实现的完整攻略：

安装CUDA

步骤1 - 下载CUDA toolkit

首先，我们需要去NVIDIA官网下载CUDA toolkit。以下是下载链接：https://developer.nvidia.com/cuda-downloads
选择适合你操作系统版本的CUDA版本，下载并保存到本地。

步骤2 - 安装CUDA toolkit

下载完成后，我们需要将其解压到目标文件夹下，然后执行安装。以下是解压命令和安装命令：

sudo sh cuda_*.run --override

执行完安装命令后，你需要配置环境变量。在终端中输入以下命令：

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

现在CUDA已安装完毕，我们可以进行一些测试来确认是否安装成功。以下是测试命令：

nvcc -V

如果安装成功，你应该可以看到CUDA版本号。

安装CUDNN

步骤1 - 下载CUDNN

进入https://developer.nvidia.com/cudnn下载CUDNN。需要注册才能下载文件。请确定下载的文件版本与CUDA版本匹配。

步骤2 - 安装CUDNN

下载文件后，按照以下步骤进行安装：

sudo dpkg -i libcudnn*.deb

注意：安装过程可能会涉及到依赖项的安装，根据提示选择安装即可。

卸载CUDA

如果需要卸载CUDA，请按以下步骤操作：

1. 删除CUDA toolkit所在的文件夹。默认情况下，CUDA toolkit会被安装在/usr/local/cuda/下。如果手动更改安装路径，则需要删除更改后的路径下的所有文件。

2. 删除CUDA相关软链接。在终端中输入以下命令：

sudo rm /usr/bin/nvcc /usr/local/cuda /usr/local/cuda-*

其中nvcc是CUDA的编译器，/usr/local/cuda/是CUDA的文件夹，/usr/local/cuda-*是CUDA的软链接。

1. 删除CUDA环境变量。在终端中输入以下命令：

sudo sed -i '/cuda/d' /etc/environment

卸载CUDNN

如果需要卸载CUDNN，请按以下步骤操作：

1. 找到libcudnn的安装路径。在终端中输入以下命令：

sudo dpkg -L libcudnn7

其中，libcudnn7是CUDNN的安装包名称。

1. 删除libcudnn相关文件。根据第1步输出的文件列表，依次使用以下命令进行删除：

sudo rm 文件路径

以我的电脑为例，删除的命令如下：

sudo rm /usr/local/cuda-10.1/include/cudnn.h
sudo rm /usr/local/cuda-10.1/lib64/libcudnn.so.7
sudo rm /usr/local/cuda-10.1/lib64/libcudnn.so.7.6.5

1. 删除CUDNN环境变量。在终端中输入以下命令：

sudo sed -i '/cudnn/d' /etc/environment

以上就是Ubuntu安装和卸载CUDA和CUDNN的完整攻略。以下是两条示例说明：

示例1 - 安装CUDA

1. 下载CUDA toolkit：https://developer.nvidia.com/cuda-downloads。
2. 将下载好的文件解压到目标文件夹下。
3. 执行以下命令安装CUDA：

sudo sh cuda_*.run --override

1. 配置环境变量。在终端中输入以下命令：

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

1. 使用以下命令测试是否安装成功：

nvcc -V

如果返回类似以下信息，则安装成功。

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Apr_24_19:10:27_PDT_2019
Cuda compilation tools, release 10.1, V10.1.168

示例2 - 卸载CUDNN

1. 利用以下命令查找CUDNN文件的安装路径：

sudo dpkg -L libcudnn7

1. 根据命令输出的文件列表，依次使用以下命令删除文件（以我的电脑为例）：

sudo rm /usr/local/cuda-10.1/include/cudnn.h
sudo rm /usr/local/cuda-10.1/lib64/libcudnn.so.7
sudo rm /usr/local/cuda-10.1/lib64/libcudnn.so.7.6.5

1. 删除CUDNN环境变量。在终端中输入以下命令：

sudo sed -i '/cudnn/d' /etc/environment

以上就是两个示例操作，并按详细步骤说明了Ubuntu安装和卸载CUDA和CUDNN的完整攻略。