文章目录
按照caffe 官方在windows环境下的要求配置caffe,生成pycaffe库
我在这里将 详细介绍如何使用 vs2015 + CMake + Ninja 的方式在win10 x64环境下配置python3.5的caffe第三方可供调用的pycaffe包。
CMake 安装与配置
在CMake官网下载安装程序,但注意版本要>=3.4。直接安装CMake,但在安装的时候要记得勾选将CMake包含在电脑的环境变量中;如果安装时没有勾选,那也没关系,直接找到CMake的安装路径,将其bin文件所在路径添加到环境变量中,如下图:
VS2015 community版 安装
直接在microsoft官网下载安装2015版本的社区版即可。
Git 安装
Git直接在官网下载win64下面的安装包直接安装即可,Git用来下载Ninja和caffe到本地电脑上。
Ninja安装与配置
Ninja的安装依赖于vs2015(或者2013),因此一定要安装好vs2015之后再来安装Ninjia。Ninja安装步骤如下:
- 打开cmd命令窗口,在某个路径下,下载Ninja包,如在C:\wq_documents\ZTE\caffe 路径下, 输入以下命令,将会联网将Ninja clone到该目录下面,然后会进入ninja子目录:
git clone git://github.com/ninja-build/ninja.git && cd ninja
- 在ninja下载完成后,以管理员的身份打开vs2015 x64 Native Tools Command(一定要是此窗口,而不能直接打开cmd窗口进行ninja编译),如下图:
- 打开该 vs2015 x64 native tools command命令窗口 后,定位到ninja所在的路径,输入下面的命令,将会编译ninja,直到完成。但是在这一步中,一定要将python添加到环境变量中,不然会报错(如果是用conda安装的虚拟python(我使用的是python3.5)环境,也可以将python.exe所在的路径添加到环境变量中,像添加CMake到环境变量中一样):
python ./configure.py --bootstrap
- 编译完成后,会出现下列东西在ninja路径当中,同时,一定要将此路径添加到环境变量中,不然到时候用CMake编译caffe会报错。
将ninja添加到环境变量中:
到这一步,我们已经准备好了编译caffe的所有前提准备了,一定要确保所有的环境变量都配置好了。 后面将介绍如何编译caffe。
caffe 编译
- 修改 scripts目录下 build_win.cmd 文件,可以让WITH_NINJA=1,表示用Ninja编译(或者直接利用vs2015 直接编译也可以, 不用安装Ninja)
- 利用Git clone caffe-windows到本地文件夹(路径随便你设置),输入以下命令即可(但是 在执行build_win.cmd之前, 一定要将build_win.cmd文件中的python版本按要求该为3,不然会出现python版本不匹配的问题 ),等待caffe编译完成,就可以了:
git clone https://github.com/BVLC/caffe.git
cd caffe
git checkout windows
scripts\build_win.cmd
- 在anaconda创建的虚拟环境下,输入一下命令,是能够使用pycaffe重要的步骤:
conda config --add channels conda-forge
conda config --add channels willyd
conda install --yes cmake ninja numpy scipy protobuf==3.1.0 six scikit-image pyyaml pydotplus graphviz
- 将编译好的caffe包拷贝至虚拟环境的site-packages中,如下图所示:
编译好的caffe python接口有以下内容