CentOS7.6安装NVIDIA驱动及CUDA、cuDNN全攻略

"这篇文档是关于在CentOS 7.6系统上成功安装NVIDIA驱动、CUDA和cuDNN的个人经验总结。适用于拥有GTX1070mobile显卡、i7-6700k处理器和16GB内存的用户。文中详细介绍了每个步骤，包括安装必要的编译环境、禁用nouveau驱动、更新initramfs以及下载并安装NVIDIA驱动。此外，还提到了NVIDIA驱动与CUDA版本的对应关系，以及安装过程中需要注意的参数选项。" 在CentOS 7.6上安装NVIDIA驱动、CUDA和cuDNN是一个重要的过程，对于需要进行GPU加速计算的开发者来说尤其关键。以下是对整个过程的详细解释： 1. **安装NVIDIA驱动前的准备** - 首先，确保系统是64位的，并且拥有与NVIDIA驱动兼容的硬件，例如GTX1070mobile显卡。 - 安装必要的编译工具，包括`gcc`，`kernel-devel`和`kernel-headers`。`kernel-devel-uname-r==$(uname -r)`确保安装的kernel-headers与当前运行的内核版本匹配，`dkms`用于动态编译模块。 2. **禁用nouveau驱动** - nouveau是开源的NVIDIA驱动，但可能与官方驱动冲突。使用`vim`编辑`/etc/modprobe.d/blacklist.conf`，添加`blacklist nouveau`，并注释掉`blacklist nvidiafb`（如果存在）。 3. **创建新的initramfs image** - 移动现有的`initramfs`文件并创建新的，这样在启动时不会加载nouveau驱动。运行`dracut`命令以更新内核映像。 4. **验证nouveau是否已禁用** - 重启后，通过`lsmod | grep nouveau`检查nouveau是否不再加载。 5. **下载并安装NVIDIA驱动** - 访问NVIDIA官方网站，根据CUDA版本选择合适的驱动。例如，CUDA 10.1对应的是NVIDIA驱动418.88版。 - 下载驱动到`/home`目录，赋予执行权限，并运行安装脚本，指定`kernel-source-path`和`--no-drm`选项，避免安装错误。 6. **安装CUDA和cuDNN** - 安装CUDA需要访问NVIDIA的CUDA Toolkit页面，下载与驱动兼容的CUDA版本。解压后，按照官方提供的安装指南进行安装。 - cuDNN是CUDA的深度学习库，同样需要从NVIDIA网站获取，下载对应CUDA版本的cuDNN，将其解压到CUDA安装目录的相应位置。 7. **环境变量配置** - 安装完成后，需要将CUDA和cuDNN的路径添加到`PATH`和`LD_LIBRARY_PATH`环境变量中，以便于系统能找到这些库。 8. **测试安装** - 运行`nvidia-smi`检查驱动是否正确安装，以及GPU的状态。 - 运行CUDA示例程序或深度学习框架（如TensorFlow或PyTorch）的简单测试，确保CUDA和cuDNN功能正常。 在安装过程中，可能会遇到各种问题，如依赖冲突、驱动不兼容等。此时，应仔细阅读错误信息，查找解决方案，或者参考NVIDIA官方文档和社区论坛中的常见问题解答。在所有组件都正确安装并配置好后，就可以充分利用NVIDIA GPU进行高性能计算和深度学习任务了。