在ubuntu下安装caffe相比起别的linux系统来,要方便的很多。用用师长的话说就是10分钟搞定。目前来看的话的确是这样的。

第一步:配置环境

两句话搞定!

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 libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install python-dev
sudo apt-get install libatlas-base-dev

第二步:安装cuda

没有cuda的caffe就是一坨屎!这句话毫不夸张,所以接下来我们要来装cuda,安装前要在BIOS设置里选择用Intel显卡来显示或作为主要显示设备:

  1. 首先运行lspci | grep -i nvidia查看自己的GPU是什么型号。
  2. 到这https://developer.nvidia.com/cuda-gpus看看能不能找到自己的型号,如果有,就继续。
  3. 然后,去nvidia官网http://www.geforce.cn/drivers选择自己的型号下载驱动。
  4. 在这https://developer.nvidia.com/cuda-downloads下载最新版本的cuda吧。
  5. root环境下编辑/etc/modprobe.d/blacklist.conf在其中添加然后保存:
    blacklist vga16fb
    blacklist nouveau
    blacklist rivafb
    blacklist nvidiafb
    blacklist rivatv
    
  6. 删除原有驱动
    sudo apt-get remove --purge nvidia-*
    sudo apt-get remove --purge xserver-xorg-video-nouveau
    
  7. ctrl+alt+F1进入没有图形界面的模式,登陆后执行sudo service lightdm stop
  8. 分别运行驱动以及CUDA即可!
  9. 将cuda库的目录添加到环境变量中export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

第三步:编译caffe:

  1. 执行git clone https://github.com/BVLC/caffe.git将最新的caffe下载下来。
  2. 执行cp Makefile.config.example Makefile.config创建一个新的makefile.config
  3. 编辑Makefile.config配置编译选项(不过基本上没啥要配置的可以先进行下一步)
  4. 编译:
    make all
    make test
    make runtest
    

    如果看到:

    [----------] Global test environment tear-down
    [==========] 1927 tests from 259 test cases ran. (248208 ms total)
    [  PASSED  ] 1927 tests.
    

    说明caffe配置成功,可以开始配置caffe下的python环境了

第四步:配置python:

切换到python目录,运行for req in $(cat requirements.txt); do pip install $req; done

有可能需要管理员权限运行for req in $(cat requirements.txt); do sudo pip install $req; done

如果安装的时候提示error: library dfftpack has Fortran sources but no Fortran compiler found参考http://www.cnblogs.com/cj695/p/5216028.html

如果提示fatal error: numpy/arrayobject.h No such file or directory参考http://www.cnblogs.com/cj695/p/5216211.html

前环岛原来caffe根目录运行make pycaffe,至此,整个caffe的配置过程到此结束!