Ubuntu16.04安装CUDA8.0+CUDNN5.1+OpenCV3.1.0+Caffe详细教程

"Ubuntu16.04安装CUDA8.0、CuDNN5.1、OpenCV3.1.0、Caffe的详细步骤，适用于单双显卡配置。" 在Ubuntu 16.04上安装CUDA、CuDNN、OpenCV和Caffe是一个涉及多个组件和依赖关系的过程。以下是一个详细的安装指南： 一．系统更新 首先，确保你的系统是最新的，以避免安装过程中遇到兼容性问题： 1. 更新软件包列表：`sudo apt-get update` 2. 升级已安装的软件包：`sudo apt-get upgrade` 二．安装依赖项 这些依赖项是构建和运行Caffe所必需的： 1. 安装protobuf、Leveldb、Snappy、OpenCV、HDF5等库的开发文件：`sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler` 2. 安装Boost库：`sudo apt-get install --no-install-recommends libboost-all-dev` 3. 安装数值计算库：`sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev` 4. 安装GFLAGS、Google Glog和LMDB：`sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev` 5. 获取Git、CMake和基本编译工具：`sudo apt-get install git cmake build-essential` 确保所有依赖项都已正确安装，可以多次执行上述命令。 三．禁用nouveau nouveau是开源的NVIDIA驱动，可能与CUDA冲突。需要禁用它： 1. 使用VIM编辑`blacklist.conf`：`sudo vim /etc/modprobe.d/blacklist.conf` 2. 在文件末尾添加： ``` blacklist vga16fb blacklist nouveau blacklist rivafb blacklist rivatv blacklist nvidiafb ``` 3. 更新内核模块：`sudo update-initramfs -u` 4. 重启系统：`sudo reboot` 5. 检查nouveau是否已被屏蔽：`lsmod | grep nouveau`，如果没有输出，则表示成功。 四．安装NVIDIA驱动 1. 首先在BIOS设置中关闭安全启动，方法因主板不同而异。 2. 添加Graphics Drivers PPA源： a. `sudo add-apt-repository ppa:graphics-drivers/ppa` b. `sudo apt-get update` 3. 查找适配的NVIDIA驱动版本：`ubuntu-drivers devices`，选择带"recommended"标记的版本。 4. 关闭图形桌面以安全安装驱动：`sudo service lightdm stop` 5. 安装NVIDIA驱动，例如版本390： `sudo apt-get install nvidia-driver-390` 五．安装CUDA 8.0 1. 下载CUDA 8.0的.run文件，访问NVIDIA官网获取。 2. 添加执行权限：`chmod +x cuda_8.0.xxxxx.run` 3. 运行安装脚本：`sudo ./cuda_8.0.xxxxx.run`，在安装过程中取消对NVIDIA 32位库的支持（如果不需要）。 4. 安装完成后，添加环境变量到`~/.bashrc`： ``` export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 5. 使修改生效：`source ~/.bashrc` 六．安装CuDNN 5.1 1. 从NVIDIA官方网站下载CuDNN的.tgz文件。 2. 解压缩文件：`tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz` 3. 将库文件复制到CUDA安装目录： a. `sudo cp -P include/cudnn.h /usr/local/cuda/include/` b. `sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64/` 4. 更新动态链接库：`sudo ldconfig` 七．安装OpenCV 3.1.0 1. 克隆OpenCV仓库：`git clone https://github.com/opencv/opencv.git` 2. 从仓库中创建并进入一个构建目录：`cd opencv && mkdir build && cd build` 3. 配置OpenCV构建：`cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_CUDA=YES -D ENABLE_AVX=YES ..` 4. 编译并安装：`make -j4 && sudo make install` 八．安装Caffe 1. 克隆Caffe仓库：`git clone https://github.com/BVLC/caffe.git` 2. 修改Caffe的Makefile.config，确保指向正确的CUDA和OpenCV路径。 3. 从Caffe目录构建：`make all -j4 && make pycaffe` 4. 安装Caffe到Python环境中：`sudo make install` 完成以上步骤后，你应该已经在Ubuntu 16.04上成功安装了CUDA 8.0、CuDNN 5.1、OpenCV 3.1.0和Caffe，现在可以开始使用它们进行深度学习项目了。记得在使用过程中遇到问题时检查日志和配置，确保所有组件都能正常工作。