Ubuntu16.04 +Caffe CPU/GPU 深度学习环境

Ubuntu16.04 +Caffe CPU/GPU 深度学习环境

最近实验室要搭建Caffe深度学习GPU开发环境，我反复在实验室电脑和自己笔记本上尝试；参考网上的众多教程后，还是出现许多莫名其妙的错误。反复重做系统十几次，终于一遍遍试出了门道。由于ubuntu系统、显卡驱动以及CUDA版本更新，之前的教程好多步骤都可以不需要做了。比如GCC版本降级问题，Caffe编译配置也都简化了很多。
所以在这里我记录下自己成功搭建好的最少步骤，希望学习Caffe的朋友少走弯路。

预先下载cuda和cudnn

安装好 ubuntu16.04 并下载cuda和cudnn安装包:
cuda_8.0.61_375.26_linux.run
https://developer.nvidia.com/cuda-downloads
cudnn-8.0-linux-x64-v5.1.tgz(官网下载需先注册一个账号)
https://developer.nvidia.com/cuda-downloads
如果只配置 CPU 版的 Caffe 或者想要在虚拟机配置 Caffe,直接参考第三步:安装配置 Caffe就可以。

本教程使用的平台:
Ubuntu16.04 64位双系统(联想笔记本)
NVIDIA 显卡 GT 755(显卡驱动版本375-39)
——教程修正时间:2017.3.14

一.安装 NVIDIA 显卡驱动

根据自己电脑 NVIDIA 显卡型号,查看支持的驱动型号。
(安装 NVIDIA 显卡驱动尽量在文本模式下,Ctrl+Alt+F1~F6)

sudo apt-get purge nvidia-*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-375
sudo reboot

重启电脑后,查看显卡驱动安装状态:

nvidia-smi

如果没出错,则驱动安装成功。

二.安装 cuda 和 cudnn

1.安装 cuda

切换到 cu