Mac OS X 10.10安装深度学习框架Caffe教程

Mac OS X 10.10安装深度学习框架Caffe教程

Author: Hoondy.com
翻译和修订:赵毅力(panovr at gmail dot com)

这份教程是关于在Mac OS X 10.10上安装深度学习框架Caffe的详细指南(成功测试的操作系统是Mac OS X10.10.3,2.3 GHz的英特尔酷睿i7 CPU,NVIDIA的GeForce GT650M显卡)

在过去的几天,我一直尝试着在我的MacBook Pro上面安装Caffe框架。Caffe是由Berkeley视觉小组开发的深度学习框架,你可以从Caffe的主页上http://caffe.berkekeyvision.org/获取更多关于它的介绍。如果你正在阅读这篇教程,你可能已经注意到了要在Mac OS X上面安装Caffe框架,你必须安装正确的依赖,并从不同的地方下载几个第三方库和工具包。Caffe网站上关于Mac OS X的安装文档有点过时,并不十分完整,有时还有点令人费解。所以,我决定分享我的成功故事,并为简单起见,我把它做成一个一步一步的安装教程,用于帮助和指导其他人如何在Mac OS X上安装Caffe框架。为了获得最大的计算性能,我将使用我的MacBook Pro上面的NVIDIA GPU并链接NVIDA的cuDNN GPU库进行加速。


详细安装步骤:
  • Homebrew
  1. 根据http://brew.sh/上面的说明安装Homebrew包管理
  • Anaconda Python
  1. 从https://store.continuum.io/cshop/anaconda/下载和安装Anaconda Python包(其中包括Caffe框架用到的hdf5)
  2. export PATH=~/anaconda/bin:$PATH
  • CUDA
  1. 从https://developer.nvidia.com/cuda-downloads下载并安装CUDA 7.0
  2. 从http://www.nvidia.com/object/mac-driver-archive.html下载并安装最新的CUDA 独立驱动
  3. export PATH=/Developer/NVIDIA/CUDA-7.0/bin:$PATH
  4. export DYLD_LIBRARY_PATH=/Developer/NVIDIA/CUDA-7.0/lib:$DYLD_LIBRARY_PATH
  • BLAS - Intel MKL
  1. 由于Mac OS X操作系统自带的BLAS库存在一些不稳定的问题,因此我选择安装Intel MKL库。如果你是在校大学生,可以使用学校邮箱从https://software.intel.com/en-us/qualify-for-free-software/student页面申请Intel Parallel XE 2015安装包(后面不要忘记在Makefile.config中设置BLAS:=MKL)
  2. 确保在安装Intel Parallel XE时选择每一个组件(因为缺省情况下不会安装MKL组件)
  3. cd /opt/intel/mkl/lib/
  4. sudo ln -s . /opt/intel/mkl/lib/intel64(因为在编译Caffe时Caffe会从MKL的intel64目录中去搜索mkl的库,但是在安装MKL后,MKL的lib目录下并没有intel64这个目录,所以需要建立一个intel64目录到lib目录的软链接)
  • cuDNN
  1. 从https://developer.nvidia.com/cudnn页面下载并安装cuDNN库(别忘了在Makefile.config中取消USE_CUDNN := 1的注释)
  2. tar -xzvf cudnn-6.5-osx-v2.tgz
  3. cd cudnn-6.5-osx-v2
  4. sudo cp lib* /usr/local/cuda/lib
  5. sudo cp cudnn.h /usr/local/cuda/include/
  • 通过Homebrew安装依赖项
  1. brew edit opencv
  2. 将下面两行
  3. args << "-DPYTHON#{py_ver}_LIBRARY=#{py_lib}/libpython2.7.#{dylib}"
  4. args << "-DPYTHON#{py_ver}_INCLUDE_DIR=#{py_prefix}/include/python2.7"
  5. 替换为
  6. args << "-DPYTHON_LIBRARY=#{py_prefix}/lib/libpython2.7.dylib"
  7. args << "-DPYTHON_INCLUDE_DIR=#{py_prefix}/include/python2.7"
  8. brew install --fresh -vd snappy leveldb gflags glog szip lmdb homebrew/science/opencv
  9. brew install --build-from-source --with-python --fresh -vd protobuf
  10. brew install --build-from-source --fresh -vd boost boost-python
  • 从Github上面克隆Caffe的代码
  1. git clone https://github.com/BVLC/caffe.git
  2. cd caffe
  3. cp Makefile.config.example Makefile.config
  • 配置Makefile.config
  1. 设置BLAS := mkl
  2. 取消USE_CUDNN := 1注释
  3. 检查并设置Python路径
  • 设置环境变量
  1. export DYLD_FALLBACK_LIBRARY_PATH=/usr/local/cuda/lib:$HOME/anaconda/lib:/usr/local/lib:/usr/lib:/opt/intel/composer_xe_2015.2.132/compiler/lib:/opt/intel/composer_xe_2015.2.132/mkl/lib
  • 编译Caffe
  1. make clean
  2. make all
  3. make test
  4. make runtest
  5. make pycaffe
  6. make distribute
原文地址:http://hoondy.com/2015/04/03/how-to-install-caffe-on-mac-os-x-10-10-for-dummies-like-me/

未经授权,请勿转载,谢谢!

评论

Shield说…
make all -j8 这样快点


What an awesome post, I just read it from start to end. Learned something new after a long time.


SAP FICO Training in Chennai

SAP SD training in Chennai

此博客中的热门博文

Some notes about building Caffe RC3 with Mac OS X 10.11.3, Anaconda, CUDA 7.5, cuDNN 4, Intel MKL and MATLAB R2015b