您的位置:首页 > 编程语言 > Python开发

Caffe-Windows + Anaconda3编译Caffe的python接口

2018-01-23 15:12 543 查看

一:目的

Caffe-Windows编译python接口;

网上找了很多资料,基本上都是用Anaconda2进行编译的,但是个人电脑中装的是Anaconda3,故这里采用Anaconda3进行编译并做下记录。

二:编译过程

1. 思路

个人电脑里装了python27,python36和Anaconda3,而Caffe-Windows编译python接口基本上都是基于python27做的,尝试直接用Anaconda3做也出现了错误,因此还是选择用Python27的环境进行编译。

主要有两个思路,一是直接用python27进行编译,也是可行的,但是在caffe-windows\python\requirements.txt文件中,可以看出需要的python包挺多的,一个个安装较为繁琐。第二个想法是在Anaconda3中建立一个python27的环境,所需的包直接用Anaconda3进行下载,相对来说方便一点,这里主要选择了第二个思路。




2. 在Anaconda3中建立一个python27的环境

(1) 在开始菜单中找到Anaconda3-Anaconda
Prompt,打开Anaconda Prompt,输入以下命令

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --set show_channel_urls yes
conda config --remove channels defaults
#从清华镜像加载,不然总是从官网下载超时。
然后添加python2.7;
conda create -n py27 python=2.7


添加完成后,打开Anaconda3,在Enviroments中可以看出已经有python27的环境了



(2) 将python27环境添加进环境变量





3. 编译python接口

(1) 修改E:\Caffe\using\caffe-windows\windows\CommonSettings.props文件



首先需要将PythonSupport设置为true,其次应当将Pythondir设置为刚才的py27路径

(2) 打开caffe-windows\windows\Caffe.sln,设置为x64 release,编译pycaff



(3) 编译成功后,在caffe-windows\Build\x64\Release目录下会生成pycaffe目录



将这个文件夹直接拷贝进Anaconda3\envs\py27\Lib\site-packages


4. 测试

(1) 打开命令行,cd进Anaconda3\envs\py27目录,因为python.exe与之前的版本重复了,这里我将其重命名了为python1.exe,输入python1(其实由于我们已经将其添加进了环境变量,不用cd进该目录,直接输入python1即可)即可打开py27交互命令行。



(2) 输入import caffe,如果能成功,就证明编译成功了。



此时可能出现一些问题,大多说是说ImportError,如ImportError: No module named google.protobuf.internal ,一般是因为缺少caffe-windows\python\requirements.txt文件中提到的包,此时可以进入Anaconda3\Enviroment\py27下进行下载,再重新编译并重复上述操作即可!

finished!
enjoy!

参考:

win下anaconda装python2.7和3.5以及jupyter共存

【caffe-Windows】微软官方caffe之 Python接口配置及图片生成实例
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息