win10 vs2015 编译caffe2

安装vs2015 cuda8 python3 cmake

下载代码

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

修改 build_windows.bat 并执行

用记事本打开CAFFE2_ROOT/scripts/build_windows.bat
在文件首的 @echo off之下添加
   set USE_CUDA=ON  （如果只是使用CPU，不使用GPU，那么这行就不要添加）
将   -DBUILD_PYTHON=OFF^  改为   -DBUILD_PYTHON=ON^
之后就可以执行 build_windows.bat，成功之后会出现提示信息： "Caffe2 built successfully"
在CAFFE2_ROOT目录下会出现 build文件夹

在cmd下运行build_windows.bat

不能用管理员权限右键打开build_windows.bat，会报路径错误。

 CMake Error at third_party/protobuf/cmake/cmake_install.cmake:39 (file):
27>    file cannot create directory: C:/Program Files/Caffe2/lib.  Maybe need
27>    administrative privileges.
27>  Call Stack (most recent call first):

27>    cmake_install.cmake:37 (include)

13. 安装Caffe2

1.先编译proto

右击 INSTALL 项目，build
完成之后会在 C:\Program Files 出现 Caffe2 文件夹
将 CAFFE2_ROOT\build\caffe2\python\Release 下的 
caffe2_pybind11_state.pyd 和 caffe2_pybind11_state_gpu.pyd 拷贝到 C:\Python27\DLLs
配置环境变量： PYTHONPATH = CAFFE2_ROOT\build 或者
PYTHONPATH = C:\Program Files\caffe2

• 1
• 2
• 3
• 4
• 5
• 6

14. 测试caffe2，跑mnist

在任意目录新建 test.py，代码如下

import numpy as np
import os
import shutil
from caffe2.python import core, cnn, net_drawer, workspace, visualize
core.GlobalInit(['caffe2', '--caffe2_log_level=0'])
caffe2_root = "~/caffe2"
print("Necessities imported!")