Ubuntu 18.04 64位下Caffe GPU安装教程：包括CUDA修复与OpenCV编译

在Ubuntu 18.04 64位系统上安装Caffe是一个常见的任务，特别是在处理深度学习和计算机视觉项目时。本篇指南详细阐述了如何在该环境下安装Caffe，包括针对可能遇到的bug进行修正以及OpenCV的编译。以下是安装步骤： 1. **更新包列表**： 首先，确保系统包是最新的，通过运行 `sudo apt-get update` 和 `sudo apt-get upgrade`。 2. **安装基本构建工具**： 安装必要的构建工具是安装Caffe的基础，使用 `sudo apt-get install build-essential`。 3. **针对CUDA的准备**： 如果你想利用GPU加速，你需要安装CUDA。在这个版本，CUDA 9.2被推荐。下载并运行CUDA的安装脚本 `cuda_9.2.88_396.26_linux.run`，注意要解决NVIDIA驱动和 Nouveau图形驱动之间的冲突。为了阻止Nouveau驱动，编辑`/etc/modprobe.d/blacklist-nouveau.conf`，添加 `blacklist nouveau` 并禁用它。 4. **更新内核模块**： 重启系统进入文本模式，检查Nouveau驱动是否已移除。如果未成功移除，手动卸载 `sudo rmrf /lib/modules/$(uname -r)/kernel/drivers/gpu/drm/nouveau/*.ko` 和备份原始文件，然后再次更新内核模块。 5. **卸载和清理NVIDIA相关软件**： 使用 `sudo apt-get remove --purge nvidia-*` 清理NVIDIA的残留组件。 6. **重新编译CUDA**： 恢复CUDA的安装，使用 `sudo ./cuda_9.2.88_396.26_linux.run`，并检查安装是否成功（通过查看`nvcc`命令是否存在）。 7. **确认CUDA版本和安装Caffe库**： 使用 `cat /proc/driver/nvidia/version` 确认CUDA版本，并通过 `sudo apt install nvidia-cuda-toolkit` 安装完整的CUDA Toolkit，这将包含CUDNN支持。 8. **编译和测试CUDA示例**： 为了验证CUDA功能，你可以尝试在Caffe样例目录下编译和运行，如 `cd NVIDIA_CUDA-9.2_Samples`，执行 `make` 和 `make install` 来编译NVIDIA提供的示例，然后运行它们来确认一切正常。 这个过程中可能会遇到一些挑战，如版本兼容性问题或驱动冲突。确保在安装前对硬件配置有清晰了解，同时遵循官方文档和社区论坛的指导，以解决可能的问题。完成这些步骤后，你将能够在Ubuntu 18.04 64位系统上顺利安装和使用Caffe进行深度学习工作。