win10下编译cpu版caffe并生成pycaffe


前言

        现在各种各样的深度学习框架层出不穷,但是说到硬件部署,很多还是离不开经典的Caffe深度学习框架,使用pytorch或者tensorflow训练完转成了onnx文件,会发现原来最难的是好装好caffe框架进行实际测试。




一、注意点

1. 尽量使用python3.5,因为主流的包还能支持,网上的很多教程是基于python2.7+vs2013的,配置python2.7的脚本倒是能够编译成功,但是后面使用发现很多如pytorch这样的库不支持。

2. 使用python3.5编译安装的话,必须配合vs2015,否则容易编译失败,虽然vs2015安装容易出现包损坏的提示报错,但是没办法。

3. 不少教程配置了Ninja进行编译,不过这个是非必须的,无非是加快编译,这个安装时候发现也会出错。



二、编译过程



1.编译准备

1.1 CMake的官网找到>=3.4的版本安装,作为工程编译工具

win10下编译cpu版caffe并生成pycaffe_第1张图片

  1.2 下载安装VS2015,根据github上的caffe源码下的提示可以看出,安装python3.5必须vs2015

win10下编译cpu版caffe并生成pycaffe_第2张图片

 1.3 安装git工具,git官网:Git - Downloads



2.编译过程

2.1 在git bash 下将caffe的源码拉下来,并切换到windows分支

git clone https://github.com/BVLC/caffe.git
cd caffe
git checkout windows

win10下编译cpu版caffe并生成pycaffe_第3张图片

2.2 设置编译文件,位置在 caffe\scripts\build_win.cmd

修改如下:其中Ninja可以改为0,不使用

win10下编译cpu版caffe并生成pycaffe_第4张图片

另外还要修改caffe/CMakeLists.txt的文件,在其中50行左右,修改为3,表明使用python3

win10下编译cpu版caffe并生成pycaffe_第5张图片

2.3 运行脚本开始编译

scripts\build_win.cmd

win10下编译cpu版caffe并生成pycaffe_第6张图片

2.4. 编译好的caffe python接口有以下内容

win10下编译cpu版caffe并生成pycaffe_第7张图片

2.5 将编译好的caffe包拷贝至python3.5安装路径的site-packages中,即将【caffe->python】文件夹下的【caffe】文件夹拷贝至【python3->Lib->site-packages】中

win10下编译cpu版caffe并生成pycaffe_第8张图片




总结

        最主要的是源码和vs要匹配,如果只是获取pycaffe的话其实也不一定需要自己编译,可以直接拷贝别人编译好的文件。提供一下我编译的方便大家使用

win10python35下编译好的pycaffe文件-Python文档类资源-CSDN下载

参考

win10+python3.5+pycharm配置caffe环境_AlisaZqq-CSDN博客

你可能感兴趣的:(深度学习,caffe)