caffe学习笔记16-caffe编译配置

本文记录了在编译Caffe时遇到的问题及其解决方案,包括make matcaffe时的错误提示以及Python导入caffe模块失败的修复方法,通过安装相关依赖库并重新编译pycaffe,成功解决了问题。
摘要由CSDN通过智能技术生成
1. 切换到Caffe-master的文件夹,生成Makefile.config配置文件,执行:
    $ cp Makefile.config.example Makefile.config

2. 配置Makefile.config文件(仅列出修改部分)
    a. 启用CUDNN,去掉"#"(目前caffe-master仍然只支持R1版本)
         USE_CUDNN := 1
   b. 启用GPU,添加注释"#"
         # CPU_ONLY := 1
   c. 配置一些引用文件(增加部分主要是解决新版本下,HDF5的路径问题)
         INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
         LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
   d. 启用Intel Parallel Studio XE 2015 Professional Edition for C++ Linux
         BLAS := mkl
   e. 配置路径,实现caffe对Python和Matlab接口的支持
         PYTHON_LIB := /usr/local/lib
         MATLAB_DIR := /usr/local/MATLAB/R2014

3. 配置Makefile文件(实现对OpenCV 3.x的支持)
    查找“Derive include and lib directories”一节,修改“LIBRARIES +=”的最后一行,增加opencv_imgcodecs
    opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs

4. 编译caffe-master,"-j8"是使用CPU的多核进行编译,可以极大地加速编译的速度,建议使用。
    $ make all -j8
    $ make test -j8
    $ make runtest -j8

5.编译Python和Matlab用到的caffe文件
    $ make pycaffe -j8
     添加环境变量 
       vim ~/.bashrc 
     将export PYTHONPATH=/home/nielsen/caffe/python:$PYTHONPATH添加到文件中。 
       source ~/.bashrc 使更改生效。 
       这样做可以在其他地方打开python,并import caffe

    $ make matcaffe -j8


在命令行输入Python;再输入import caffe时,可能会报以下错误:
can not find module skimage.io
此时只要按照以下命令操作即可:
$ sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags Cython ipython

$ sudo apt-get update
在caffe-master目录下:
$ make pycaffe
然后在命令行输入python;再输入import caffe就可以成功啦。




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值