Caffe win10+VS2019环境配置及python接口配置
在win10+vs2019环境下成功编译运行了caffe-windows(cpu-only)
并配置好了编译生成的python接口的使用
开发环境
- 系统:Win10
- IDE:VS2019
Caffe环境配置及编译
参考:https://blog.csdn.net/waji2000/article/details/105155198
所需下载项:https://pan.baidu.com/s/19mmhrDkqKS4FjBmh7oPhQg 提取码:dv4f
caffe环境配置的难点就是有很多依赖库要装,原来如果找不到依赖库会自动下载,放到vs2019容易出错版本不匹配等等。这个博客的作者提供了bulider解决的这个问题。
这样过程就很简单了,只需要
- 配置编译好bulider,解决依赖库的问题,将原配置文件里自动下载的部分删掉,换为从bulider路径里获取。
- cmake配置生成vs2019解决方案(python路径我换成了自己anaconda下的python2.7)
- 打开vs2019编译即可
编译
vs基本的编译模式分为release和debug。
两者生成的文件名称不同,debug的文件名会多一个-d。
例如debug编译生成caffe-d.exe
,release编译生成caffe.exe
生成的python接口名字也不同,debug下文件名为_caffe-d.pyd
,release为_caffe.pyd
运行的时候要注意这点。
测试
到release目录下运行生成的exe看看(anaconda多版本python记得切换到对应的环境运行,比如我的默认环境为python3直接运行会报缺少python27.dll)
ps:我cmake生成解决方案的路径是caffe-windows\bulid
python接口配置
编译完成后的python接口存储在..\caffe-windows\python\
路径下的caffe
文件夹中。
将整个文件夹复制到所使用python环境根目录..\Lib\site-packages\
下即可。
用python运行import caffe
测试
可能会报某些库缺失,例如scipy、google protobuff等等,根据提示pip install
依次安装好。
然后就可以在自己的python环境里使用caffe的python接口了。
cmd测试draw_net.py
> cd 你的路径/caffe-windows/python
> python draw_net.py ../models/bvlc_reference_caffenet/deploy.prototxt ../models/bvlc_reference_caffenet/structure_deploy.jpg
会提示缺失库需要conda install graphviz
和conda install pydotplus
生成的caffenet结构图如下