Ubuntu 16.04下安装Caffe（GPU版本 GTX970）

2018年8月6日更新说明：距上次安装成功没多久，显卡驱动莫名失效，没有解决，索性重装了系统，考虑到还会用到Tensorflow，所以本次按照cuda9.0 + cudnn7的配置安装。原打算用anaconda来进行python编程的，但是这会影响到ROS的使用，所以暂时放弃，决定使用pycharm编程。

首先需要感谢如下博客：

https://blog.csdn.net/yhaolpz/article/details/71375762（opencv、caffe安装等参考）

https://blog.csdn.net/a_z666666/article/details/72853346（opencv、caffe安装等参考）

https://blog.csdn.net/ghw15221836342/article/details/79571559 （GPU安装参考）

https://www.cnblogs.com/5211314jackrose/p/7081020.html （GPU安装参考）

https://blog.csdn.net/babytang008/article/details/78631776 （anaconda配置参考）

https://blog.csdn.net/balixiaxuetian/article/details/79154013 （opencv3.4.1参考）

鼓捣了一天，参照网上各路大神的教程，终于安装好了，为了自己也为了他人，特此记录下来。

（因为很多都是转载，所以排版很乱）

安装硬件：Intel i7 + NVIDIA GTX 970（单显卡）

安装流程细分为如下12个步骤：

 

1、安装依赖包

2、禁用 nouveau

3、配置环境变量

4、安装显卡驱动

5、安装 CUDA 9.0

6、验证 CUDA 9.0是否安装成功 

7、安装 cudnn 

8、安装 opencv3.4.1

9、安装 caffe 

10、安装 pycaffe notebook 接口环境

11、MNIST数据集测试 

12、Nvidia显卡驱动、cuda、cudnn重新安装

 

1、安装依赖包 

安装后续步骤或环境必需的依赖包，依次输入以下命令:

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install git cmake build-essential

有一定几率安装失败而导致后续步骤出现问题，所以要确保以上依赖包都已安装成功，验证方法就是重新运行安装命令，如验证 git cmake build-essential是否安装成功，则再次运行以下命令，成功的话会提示已安装：

sudo apt-get install git cmake build-essential

 

2 、禁用 nouveau

安装好依赖包后需要禁用 nouveau，只有在禁用掉 nouveau 后才能顺利安装 NVIDIA 显卡驱动，禁用方法就是在 /etc/modprobe.d/blacklist-nouveau.conf 文件中添加一条禁用命令，首先需要打开该文件，通过以下命令打开：

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

打开后发现该文件中没有任何内容，写入：

blacklist nouveau
options nouveau modeset=0

保存时命令窗口可能会出现以下提示：

** (gedit:4243): WARNING **: Set document metadata failed: 不支持设置属性 metadata::gedit-position

无视此提示～，保存后关闭文件，注意此时还需执行以下命令使禁用 nouveau 真正生效：

sudo update-initramfs -u

电脑重启之后执行

lsmod | grep nouveau  #没有输出，即说明安装成功

 

3、配置环境变量

同样使用 gedit 命令打开配置文件：

sudo gedit ~/.bashrc

打开后在文件最后加入以下两行内容：

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH

保存退出。

source ~/.bashrc

 

4、安装显卡驱动（9.0）

        cuda中本身包含显卡驱动，但是我试着装了一下cuda9.0，会报错：“The driver installation is unable to locate the kernel source....”，为了解决这个问题特地去查找“怎么降低Ubuntu16.04的内核版本”，我的系统默认的是4.13.45，而cuda文档里要求是4.4.0版本，试了半天没解决降版本的问题，索性就不试了，先单独安装显卡驱动了。

手动去官网下载.run文件自己安装

下载完成之后：

卸载原先的所有驱动：

sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-390.59.run --uninstall

禁用nouveau（前面已做）

 

安装驱动

进入命令行界面

Ctrl-Alt+F1  之后输入用户名和密码登录即可。Ctrl-Alt+F7 退出命令行界面

sudo service lightdm stop（关闭图形界面，此时Ctrl-Alt+F7无法返回界面，除非 sudo service lightdm stop）

给驱动run文件赋予执行权限（安装文件一般放在home根目录下）

sudo chmod a+x NVIDIA-Linux-x86_64-390.59.run

sudo ./NVIDIA-Linux-x86_64-3