ubuntu15和16下安装cuda以及caffe深度学习环境

之前在台式机上搭建了一次ubuntu下的caffe深度学习环境，用的是ubutnu14.04,跟着网上的教程走，基本没遇到啥大问题，重装了一次ubuntu就把所有环境搭建好了，总共花时间不到两天。但是后面做实验发现显存不够，于是又在自己笔记本上搭建了一次ubuntu的深度学习环境，由于笔记本比较新，加上是集显加独显的，装ubuntu14的时候遇到了各种问题，无奈换了ubuntu15，虽然环境是搭好了，但是每次关机重启不是黑屏就是卡在登陆界面，要重新启动几次才能进入系统，索性一口气直接升级16了，目前编译caffe以及跑各种代码都没问题。下面是具体安装过程，有些大多数教程上已经很详细了，我就不具体叙述，只是把需要注意的部分和有坑的部分列出来。

具体安装过程

ubuntu14下的caffe安装

因为ubuntu14也安装过，这里把ubuntu14下的安装也简要說一下，建议参考博客：http://geek.csdn.net/news/detail/77859

上面的每一步流程，包括每一步测试都十分详细，不过有一个需要注意的问题就是，如果是台式机下的独显的话，千万别按照他说的先装驱动，因为一装驱动再重启就会黑屏进不了系统！虽然网上也有解决方案，但是这样一来就变得很麻烦，最方便的方法是不安装驱动，直接安装下载的cuda的deb文件，这样安装cuda的时候会自动安装驱动，而且也不会有前面的黑屏问题。

建议非独显还是不要用ubuntu14了，坑太多！

ubuntu15和ubuntu16下caffe环境的搭建

ubuntu15和16下的安装是一样的，因此在这统一叙述，建议还是直接安装ubuntu16,因为我这安装好15下的caffe环境后开机总是有点莫名其妙的问题，经常要重启几次才能正常开机。

我的笔记本环境：

双显卡：intel的集成显卡加上nvidia的GT940M 4G（双显卡在安装驱动时是个大坑，切记！）

在笔记本上安装ubuntu下的caffe环境足足花了我五六天时间，几乎把网上所有的教程都看了一遍，大部分坑都踩了一回。这里总结一条目前看来应该三最简单的路。

大部分过程可以参考博客：http://blog.csdn.net/g0m3e/article/details/51420565

此处我就不详细叙述流程，我把我安装时遇到的一些问题和经验说一下：

1.安装驱动

安装好ubuntu后第一次进入系统时，按“ctrl+T”进入命令行，输入“nvidia-settings”

然后会弹出这样一个英伟达显卡的窗口



==忽略上面的显卡型号，因为这是在台式机上写的，这图形界面会有一栏是用来选择显卡的，一般刚打开的时候默认是intel的集成显卡，需要自己手动选择另外一个nvidia显卡

这里会需要输入密码。然后点击“设置->软件和更新->附加驱动”不要选他预置的X开头的驱动，选择其他几个专有驱动，只要大于352就行，一般选最新的。具体如下图



然后重启，再次进入系统的时候查看一下“nvidia-settings”是否已经改为选中nvidia显卡。如果是的话就可以开始安装cuda了。

2.安装cuda和cudnn

cuda建议安装cuda7.5，cudnn建议安装V4,因为不同的程序会对caffe做一些拓展，然后很多代码需要单独编译自己的caffe，在这个过程中cudnn版本的问题就会凸显出来。

当然，如果你第一次编译caffe的时候成功了的话，后面再编译caffe其实只要把原先的cudnn给删除，添加另外一个版本的cudnn进去，然后重新编译就行。

这个过程前面推荐的那个教程很详细，按步骤来的话基本没啥问题，这里就不仔细叙述了。只是需要提醒一点，一定要记着/usr/local/cuda这个地址，用.run文件安装的话三默认这个地址的，这个是cuda和cudnn的地址，在有些教程上这个地址可能会不一样，千万不要看到命令行的代码就直接粘贴！

安装完cuda和cudnn之后，可以通过“nvcc -V”和“nvidia-smi”来测试一下英伟达显卡是否识别正常(没有报错即为正常）。如下图：

3.安装编译caffe

这里应该还有一些依赖项什么的，直接按照前面的教程来就好。

在Makefile.config那需要注意的是，大多数用python来写的demo都会要求用python_layer，所以最好把“# WITH_PYTHON_LAYER := 1”前面的#去掉，

还有一个就是有一个makefile文件一定要记得改！（不是makefile.config！别搞混了，它们在同一文件夹下），因为这涉及到gcc编译器版本的问题。

还有就是新手估计都不懂vim的用法，所以能够gedit的地方一定用gedit来编辑！

此外，还有一个就是如果在make的时候（如make all ，make test等）遇到报错，不妨在make前面加个sudo，即“sudo make all -j4"

我编译的时候很多次遇到这种情况，直接make报错，加上sudo之后就好了。

特别提示：编译好caffe和pycaffe之后只有在python文件夹下输入python，import caffe才不会报错说”找不到caffe模块“

这是因为没有设置环境变量的原因，这里需要人为地设置caffe的路径到环境变量中，

这里主要涉及到两个文件：/etc/profile 和 ~/.bashrc

通过在命令行输入：sudo gedit /etc/profile和sudo gedit 

~/.bashrc来进行编辑

先尝试通过命令行将Caffe添加到环境变量中，在caffe文件的根目录下输入

echo 'export CAFFE_ROOT=$(pwd)' >> ~/.bashrc
echo 'export PYTHONPATH=$CAFFE_ROOT/python:$PYTHONPATH' >> ~/.bashrc
source ~/.bashrc

然后重启，打开命令行输入python，然后import caffe

如果不报错说明成功加载到环境了，如果报错则通过前面说的用gedit进行编辑手动添加路径，格式同

export PYTHONPATH=$CAFFE_ROOT/python:$PYTHONPATH

。

后记：如果第一次编译caffe成功了，后面再编译其他程序自带的caffe的时候却失败的情况，通常都是cudnn的原因，可以尝试删除原来的cudnn文件重新添加并连接，然后再编译。还有就是编译失败多试试sudo。