深度学习|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.2 下载安装VS2015,根据github上的caffe源码下的提示可以看出,安装python3.5必须vs2015
深度学习|win10下编译cpu版caffe并生成pycaffe
文章图片

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
文章图片

2.2 设置编译文件,位置在 caffe\scripts\build_win.cmd
修改如下:其中Ninja可以改为0,不使用
深度学习|win10下编译cpu版caffe并生成pycaffe
文章图片

另外还要修改caffe/CMakeLists.txt的文件,在其中50行左右,修改为3,表明使用python3
深度学习|win10下编译cpu版caffe并生成pycaffe
文章图片

2.3 运行脚本开始编译
scripts\build_win.cmd

深度学习|win10下编译cpu版caffe并生成pycaffe
文章图片

2.4. 编译好的caffe python接口有以下内容
深度学习|win10下编译cpu版caffe并生成pycaffe
文章图片

2.5 将编译好的caffe包拷贝至python3.5安装路径的site-packages中,即将【caffe->python】文件夹下的【caffe】文件夹拷贝至【python3->Lib->site-packages】中
深度学习|win10下编译cpu版caffe并生成pycaffe
文章图片



总结 最主要的是源码和vs要匹配,如果只是获取pycaffe的话其实也不一定需要自己编译,可以直接拷贝别人编译好的文件。提供一下我编译的方便大家使用
win10python35下编译好的pycaffe文件-Python文档类资源-CSDN下载
参考
【深度学习|win10下编译cpu版caffe并生成pycaffe】win10+python3.5+pycharm配置caffe环境_AlisaZqq-CSDN博客

    推荐阅读