我爱编程

Window安装并使用Caffe2

2018-05-08  本文已影响401人  andyhacker

Caffe2是对Caffe的升级和重构,用大神们的话来说,Caffe2更多地向TensorFlow靠拢,包括Operator和WorkSpace的使用,这里就不对此多做解释。由于需要在Window上搭建一个学习环境,而学习Caffe2最好的办法就是官网文档,相当全。

Caffe2 Tutorials Overview | Caffe2

在Windows上安装Caffe2库支持大致分为一下几部(我以Python为主要编程语言为例)

1. 安装依赖库

2. 下载源码

3. 编译

4. 测试

1. 安装依赖库

由于Caffe的部分运算需要一些其他运算库支持,在安装Python之前最好先把官方建议的依赖库安装好

pip install future ^

            hypothesis ^

            numpy ^

            protobuf ^

            six

还有官方建议的可选库

pip install flask ^

            glog ^

            graphviz ^

            jupyter ^

            matplotlib ^

            pydot python-nvd3 ^

            pyyaml ^

            requests ^

            scikit-image ^

            scipy ^

            setuptools ^

            tornado

正如官网所说:

Anaconda users: If you’re using Anaconda, use conda install instead of pip install.

2. 下载源码

推荐使用git下载,没有为什么,习惯而已

git clone --recursive https://github.com/pytorch/pytorch.git

官方已经把windows安装的编译器参数都配置好了,不过前提是你已经安装好了VS2015/VS2017

在编译之前,修改一下编译配置,将下载源码文件夹中的 scripts/build_windows.bat 用编辑器打开,修改 BUILD_PYTHON = OFF为

BUILD_PYTHON = OFF

If you want to build Python bindings, you need to edit ‘\scripts\build_windows.bat’ and set BUILD_PYTHON to ON

3. 编译安装

接下来就是重头戏编译了,为了一次可以编译成功,最好把你系统里边python27.lib拷贝到pytorch文件夹下面,为了保险起见,最好以下3个文件夹下面都拷贝一份

pytorch/

pytorch/caffe2/

pytorch/build/

(本文不考虑使用GPU的情况,如有需要GPU的同学可参考官方文档)

4. 测试

编译完成之后,切换到目录\pytorch\build\caffe2\python下,拷贝 caffe2_pybind11_state.pyd 到python库目录 ...\Anaconda2\DLLs

再切换到目录 pytorch/build下,可以通过 import caffe2来测试是否安装成功

本人使用时,发现只有在上述目录下才能import成功,一气之下,将 bulid/caffe2 整个目录全部拷贝到 .../Anaconda2/Libs/sites-package/目录下面,可以使用

参考:一些错误参考

from past.builtins import xrange ImportError: No module named 'past' · Issue #12 · carpedm20/MemN2N-tensorflow · GitHub

上一篇下一篇

猜你喜欢

热点阅读