Caffe 安装 Ubuntu14.04+CUDA7.0/7.5（亲测有效）

　　自己安装Caffe的过程中，参考了很多资料，但由于版本或者其他原因，安装过程中总是遇到这样或者那样的问题，因此留做记录，方便之后查看，也希望对遇到相似麻烦的朋友们提供帮助。

　下面我们开始安装吧：

　　硬件自然不用说了，一个好的显卡（Nvida TiTan 系列），i7的处理器，8G以上的RAM。

1. 安装部分支持库

sudo apt-get install build-essential
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler 
sudo apt-get install freeglut3-dev libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

2. 下载NVIDIA显卡驱动和 CUDA

　　a. 显卡驱动下载

　　http://www.nvidia.cn/Download/index.aspx?lang=cn 选择对应型号和系统下载（CUDA向下兼容，可以下载最新的driver）

　　Linux下run文件点下载键直接打开网页，可点击右键另存为即可。

　　b. CUDA下载

　　https://developer.nvidia.com/cuda-downloads 选择对应系统下载

　　可以下载CUDA 7.0或者 7.5

3. 安装NVIDIA显卡驱动和CUDA

　　首先进入tty，输入用户名和密码。

　　关闭图形界面：

sudo stop lightdm

　　安装驱动：（更改驱动名称）

sudo chmod -R 777 NVIDIA-Linux-x86_64-352.21.run          
sudo sh NVIDIA-Linux-x86_64-352.21.run

　　安装CUDA：

sudo chmod -R 777 cuda_7.0.28_linux.run
sudo ./cuda_7.0.28_linux.run

　　安装过程都默认路径就好。CUDA7会提示安装完了Driver、Toolkit和Samples。

　　　　备注1：如果之前你单独安装了驱动，安装CUDA时，驱动安装部分可以跳过。

　　　　备注2：安装Samples可能会提示缺少一些库，simple 也可之后编译测试。　

　　打开图形界面

sudo start lightdm

4. 如果安装驱动不成功，很可能会进入黑屏模式，不要着急重装系统，进入tty重新装一遍驱动即可。（驱动安装成功，本步可略过）

　　如果进入不了tty (ctrl+Alt+F1~F6)，只是黑屏，则：

sudo sed -i -e 's/#GRUB_TERMINAL/GRUB_TERMINAL/g' /etc/default/grub
sudo update-grub
sudo reboot

　　如果重启机子之后就进入不了桌面，在登陆界面循环，但是可以进入tty，一般是内核引导或者驱动安装的问题。则进入tty

lsmod | grep nouveau

　　创建文件，不会用vi的可以用gedit

sudo gedit /etc/modprobe.d/nvidia-graphics-drivers.conf

　　写入 blacklist nouveau，退出保存（wq）

　　修改grub：

sudo gedit /etc/default/grub

　末尾写入　

rdblacklist=nouveau nouveau.modeset=0

5. 安装配置Nvidia cuDNN-v4 （也可下载v5版本，链接库文件时注意改名）

　　a. 去官网下载最新的cuDNN (cudnn-70-linux-x64-v4)。

https://developer.nvidia.com/cudnn 下载，需要先注册为developer，审核通过就可以下载，大概审个两天就可以。这个主要是针对Deep Learning框架设计的NVIDIA CUDA的GPU计算加速方案。不安装这个也可以跑caffe。

　　b. 添加头文件和库文件

sudo cp include/cudnn.h /usr/local/include
sudo cp lib64/libcudnn.* /usr/local/lib

　　c. 链接cuDNN的库文件

sudo ln -sf /usr/local/lib/libcudnn.so.4.0.7 /usr/local/lib/libcudnn.so.4
sudo ln -sf /usr/local/lib/libcudnn.so.4 /usr/local/lib/libcudnn.so
sudo ldconfig -v

6. 设置CUDA的环境变量　

　　安装完成后需要在/etc/profile中添加环境变量

sudo gedit /etc/profile

　　在文件最后添加:

PATH=/usr/local/cuda/bin:$PATH
export PATH

　　保存后, 执行下列命令, 使环境变量立即生效

source /etc/profile

　　同时需要添加lib库路径：在 /etc/ld.so.conf.d/加入文件cuda.conf，

sudo gedit /etc/ld.so.conf.d/cuda.conf

　　添加内容：

/usr/local/cuda/lib64

　　保存后，执行下列命令使之立刻生效：

sudo ldconfig

7. 安装测试CUDA simples　　

cd /usr/local/cuda/samples
sudo make all -j8

　　其中-jx中x的数字代表几个线程同时编译，可以加快速度，x可以根据CPU核的个数决定。

可能遇到的问题：/usr/bin/ld: cannot find -lGL

原因：1. 缺少libGL库文件；2. link的问题。

方案：

1，如果是缺少libGL库文件，可以先搜寻库文件安装包

apt-cache search libGL-dev
sudo apt-get install libGL-dev

2，先寻找link文件(.so)，再根据寻找结果定位　　

locate libGL.so

　　显示结果如下：

/usr/lib/x86_64-linux-gnu/libGL.so
/usr/lib/x86_64-linux-gnu/libGL.so.1
/usr/lib/x86_64-linux-gnu/libGL.so.346.46
/usr/lib/x86_64-linux-gnu/mesa/libGL.so

　　链接文件：

sudo rm /usr/lib/x86_64-linux-gnu/libGL.so
sudo ln -s /usr/lib/x86_64-linux-gnu/libGL.so.346.46 /usr/lib/x86_64-linux-gnu/libGL.so

　　其他问题解决方案类似。

　　编译完成后进入samples/bin/x86_64/linux/release, 运行deviceQuery

cd ./bin/x86_64/linux/release
./deviceQuery

　　如果出现显卡信息，则驱动及显卡安装成功。　　

8. 安装Intel MKL或Atlas

　　如果没有Intel MKL，可以用下列命令安装免费的atlas（我在安装过程中，使用MKL时，与caffe不兼容，于是直接使用了atlas！！！！）

sudo apt-get install libatlas-base-dev

当然使用Intel MKL会快一些，我在caffe社区也反映了兼容问题，现在已经解决。安装Intel MKL过程如下：

　　a. 下载Intel MKL

　　Intel MKL可以申请学生账号下载免费intelMKL　　　

　　b. 解压安装包，下面有一个install_GUI.sh文件，执行该文件sudo run intall_GUI.sh，会出现图形安装界面，按说明安装即可。

　　c. 安装完成后需要添加library路径, 创建/etc/ld.so.conf.d/intel_mkl.conf文件

sudo vi /etc/ld.so.conf.d/intel_mkl.conf

　　在文件中添加内容:　

/opt/intel/lib
/opt/intel/mkl/lib/intel64

　　注意把路径替换成自己的安装路径

sudo ldconfig

9. OpenCV的下载安装

　　a. 下载OpenCV（建议2.4 系列，3系列差别比较大）

　　在https://github.com/bearpaw/Install-OpenCV 下载到本地。opencv版本2.4.10.

　　b. 安装OpenCV

sudo sh ./opencv_2_4_10.sh

　　脚本会自动安装依赖项，下载安装包，编译并安装OpenCV。

10. 安装Caffe所需要的Python环境

　　a. 直接用系统自带的python就行了，可以通过以下命令查看python版本

python --version

　确定使用的是anaconda提供的python环境,安装

which python
which pip
apt-get install python-pip

　　在用户自己的`~/.bashrc` 中添加library path

LD_LIBRARY_PATH=your_anaconda_path/lib:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH

11. 安装matlab　　

　　下载matlab后，安装过程按照安装包里的要求做就可以了。（R2014a）

sudo ./install

　　备注1：安装完之后需要添加桌面图标，不然每次都需要到/usr/local/MATLAB/R2014a/bin中去 sudo ./matlab 。

　　备注2：不要直接去软件中心下载matlab-support,尝试过，一直在installing，没有成功。

　　备注3：MATLAB R2013a安装包，遇到过一堆乱七八糟其他问题，比如撞到83%说某个库不是exe什么的，最后换了个安装包和版本，一切顺利。

　　备注4: ubuntu和mac下，对matlab进行激活时，输入用户名和密码，总是显示连接不到服务器，我们需要将防火墙（farewall）关掉。输入“sudo ufw disable”，然后进行激活，激活完成，"sudo ufw enable" 重启防火墙即可！

12. 其他依赖项安装

　　a. Google Logging Library（glog），下载地址：https://code.google.com/p/google-glog/，然后解压安装：

tar zxvf glog-0.3.3.tar.gz
./configure
make
sudo make install

　　如果没有权限就chmod a+x glog-0.3.3 -R , 或者索性 chmod 777 glog-0.3.3 -R

　　b. 其他依赖项，确保都成功

sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler protobuf-c-compiler protobuf-compiler

13. 编译Caffe（终于到了激动人心的一刻）

　　进入caffe根目录，首先复制一份`Makefile.config`, 然后修改里面的内容，主要需要修改的参数包括

CPU_ONLY 是否只使用CPU模式，没有GPU没安装CUDA的同学可以打开这个选项
BLAS (使用intel mkl还是atlas)，建议使用atlas，方便又快捷
MATLAB_DIR 如果需要使用MATLAB wrapper的同学需要指定matlab的安装路径, 如我的路径为/usr/local/MATLAB/R2014a (注意该目录下需要包含bin文件夹，bin文件夹里应该包含mex二进制程序)
DEBUG 是否使用debug模式，打开此选项则可以在eclipse或者NSight中debug程序

完成设置后，开始编译

make all -j8
make test
make runtest

　　如果需要matlab wrapper

make matcaffe

　　如果需要python wrapper

make pycaffe

　　备注：如果有多个caffe包，一定要输出每一个的python路径。例如有caffe1和caffe2：

export PYTHONPATH=your_caffe_path/caffe1/python:$PYTHONPATH
export PYTHONPATH=your_caffe_path/caffe2/python:$PYTHONPATH

14. 使用MNIST数据集进行测试

　　Caffe默认情况会安装在$CAFFE_ROOT，就是解压到那个目录，例如：$ home/username/caffe-master，所以下面的工作，默认已经切换到了该工作目录。

　　下面的工作主要是，用于测试Caffe是否工作正常，不做详细评估。具体设置请参考官网：http://caffe.berkeleyvision.org/gathered/examples/mnist.html

　　a. 数据预处理

sh data/mnist/get_mnist.sh

　　b. 重建lmdb文件。Caffe支持多种数据格式输入网络，包括Image(.jpg, .png等)，leveldb，lmdb，HDF5等，根据自己需要选择不同输入吧。

sh examples/mnist/create_mnist.sh

　　生成mnist-train-lmdb 和 mnist-train-lmdb文件夹，这里包含了lmdb格式的数据集

　　c. 训练mnist

sh examples/mnist/train_lenet.sh

　　至此，Caffe安装的所有步骤完结，下面是一组简单的数据对比，实验来源于MNIST数据集，主要是考察一下不同系统下CPU和GPU的性能。可以看到明显的差别了，虽然MNIST数据集很简单，相信复杂得数据集，差别会更大，Ubuntu+GPU是唯一的选择了。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：Caffe 安装 Ubuntu14.04+CUDA7.0/7.5（亲测有效） - Python技术站

Caffe 安装 Ubuntu14.04+CUDA7.0/7.5（亲测有效）

相关文章