caffe python3.5 Windows版本编译

之前编译的是caffe python2.7版本,与caffe python3.5版本还是有差异,记录一下。

下载 caffe windows版本:https://github.com/BVLC/caffe/tree/windows

前提:已安装 anaconda 3(python 3.5),VS2015,CMake,CUDA8.0+cudnn5.1

一 caffe-windows\scripts路径下修改配置文件build_win.cmd

如上图, 根据提示VS、Python版本修改为相应的版本号(MSVC_VERSION=14:VS2015;PYTHON_VERSION=3:Python3.5);CPU_ONLY=0:GPU。

修改好build_win.cmd文件,可以打开CMake进行配置生成caffe的VS项目了(也可直接输入命令.\scripts\build_win.cmd)

CMake中,如图1所示,source code位置为caffe路径,build为新建的生成文件路径,下图红圈圈处BLAS其Value由Atlas改为Open,否则会出现“Could NOT find Atlas...”的错误。点击Configure,选择相应的编译器(此处VS2015),CMake会下载相应的依赖包libraries_v140_x64_py35_1.1.0.tar.bz2到C:\Users\XX\.caffe\dependencies\download下,编译时会到dependencies找相应的包(例如BLAS会调用openblas的包),配置成功,再点击Generate,完成。得到caffe的解决方案Caffe.sln。

    

                                         图1  CMake配置节目

                         图2   编译器选择

可能出现的问题:本地存在python多个版本,比如其他应用安装了python 2.7,configure时先识别python 2.7之后,会在dependencies自动下载对应python 2.7版本的依赖包libraries_v120_x64_py27_1.1.0.tar.bz2,可以先屏蔽安装python 2.7版本的应用。

     图3   configure界面

三 VS编译caffe

以ALL_BUILD为启动项,配置:Release+x64,编译整个工程,完成。

可以在 caffe-windows\build\tools\Release 下调用相应的可执行文件了

四 python中调用caffe

Caffe.sln中pycaffe编译好之后,在caffe-windows\python目录下会生成caffe文件夹,将其拷贝到Anaconda3\Lib\site-packages目录下。此时在Spyder中调用应能成功调用caffe。

然而:import caffe 时出现了 from numpy.lib.arraypad import _validate_lengths, ImportError: cannot import name '_validate_lengths'的错误,应该是scikit-image包的问题,对其更新 pip install --upgrade scikit-image之后,又出现caffe cannot import name 'img_as_float32',直接pip uninstall 再install scikit-image,此时import caffe成功。

  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值