栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

win10下编译cpu版caffe并生成pycaffe

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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的版本安装,作为工程编译工具

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

 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

2.2 设置编译文件,位置在 caffescriptsbuild_win.cmd

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

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

2.3 运行脚本开始编译

scriptsbuild_win.cmd

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

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




总结

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

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

参考

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/272568.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号