[关闭]
@hshustc 2015-10-12T22:00:19.000000Z 字数 1433 阅读 2382

Ubuntu14.04 64bit+Caffe配置过程

Linux


注:默认取得超级用户权限和设置网络状况良好

0. 开发依赖包

  1. sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev
  2. sudo apt-get install --no-install-recommends libboost-all-dev

1. cuda安装(For GPU Mode)

推荐cuda7.0,文件夹内有安装包cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb,

  1. init 3
  2. dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb
  3. apt-get update
  4. apt-get install cuda
  5. reboot

重启之后需要设置环境变量,在/etc/profile文件最后添加

  1. PATH=/usr/local/cuda/bin:$PATH
  2. export PATH

/etc/ld.so.conf.d加入文件cuda.conf,内容如下

  1. /usr/local/cuda/lib64
  2. /usr/local/cuda/lib

保存后执行

  1. ldconfig

使之立即生效

运行示例测试安装成功
进入/usr/local/cuda/samples,执行

  1. make all -j 12(12为并行线程数,推荐根据CPU核心数设置)

进行编译,编译完成后进入sample/bin/x86_64/linux/release执行

  1. ./deviceQuery

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

2.opencv安装

推荐通过脚本安装OpenCV2.4.10,文件夹内有Install-OpenCV

  1. cd Install-OpenCV/Ubuntu/2.4
  2. ./opencv2_4_10.sh

因为opencv-2.4.10.zip已经下好了,所以会节省不少时间

3.blas安装

  1. apt-get install libatlas-base-dev

4.python安装(可选)

Ubuntu14.04 64bit默认的python版本是2.7,为caffe推荐的版本,不用进行升级。
首先安装pip

  1. apt-get install python-pip python-dev build-essential
  2. cd /path/to/caffe/python
  3. for req in $(cat requirements.txt); do pip install $req; done
  4. export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

5. MATLAB安装(可选)

需保证mex路径包含在$PATH里面

6. 其他依赖项

  1. sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

7.编译caffe

  1. cd /path/to/caffe
  2. cp Makefile.config.example Makefile.config
  3. #Adjust Makefile.config(里面有详细说明每个变量该如何配置,一步步来就可以了, 文件夹中Makefile.config是一个示例)
  4. make all -j 12(12为并行线程数,推荐根据CPU核心数设置)
  5. make test -j 12
  6. make runtest -j 12
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注