Ubuntu上CUDA与CUDNN的安装与卸载指南

本文主要介绍了在Ubuntu操作系统上安装和卸载CUDA (Compute Unified Device Architecture) 和 CUDNN (CUDA Deep Neural Network Library) 的详细步骤。CUDA是NVIDIA提供的一种并行计算平台和编程模型，而CUDNN则加速了深度学习在GPU上的运算。作者的目的是在学习和实践PaddlePaddle框架时，确保显卡驱动和库的正确配置。 首先，文章强调了在安装新版本CUDA（如CUDA 10.0）和CUDNN（如CUDNN 7.4.2）之前，需要先禁用可能导致冲突的NVIDIA Nouveau显卡驱动。在Ubuntu中，通过编辑`/etc/modprobe.d/blacklist.conf`文件，添加`blacklist nouveau`和`options nouveau modeset=0`来阻止Nouveau驱动加载。接着，运行`sudo update-initramfs -u`更新内核模块并重启系统，以验证禁用操作的有效性。 接下来，作者指导读者从NVIDIA官方网站下载适合自己显卡（例如RTX 2070）的驱动程序，如`NVIDIA-Linux-x86_64-410.93.run`。卸载旧驱动涉及禁用X-Window服务，避免干扰安装过程，然后通过`apt-get remove --purge nvidia*`等命令删除旧的CUDA和驱动程序，以及手动解压并运行卸载脚本来彻底清除。 安装新驱动时，只需执行下载的驱动文件并按照提示进行，最后重新启动系统。然而，如果遇到系统重启后重复登录，可能是驱动版本不匹配，需确认并重新安装正确的驱动。 对于CUDA的卸载，作者解释了这样做的原因，即在升级或更换显卡驱动时，可能需要清理旧版本的CUDA环境，以避免潜在的冲突和资源占用。卸载CUDA通常包括删除CUDA相关软件包、卸载CUDA Toolkit和相关的环境变量设置。 总结来说，本文提供了一套完整的在Ubuntu环境下管理和优化CUDA及CUDNN的教程，对于开发人员在处理深度学习工作负载时，理解和操作显卡硬件及其相关软件库具有实际指导价值。