caffe2跟老版caffe相比变化较大,更容易上手和使用。但caffe2的windows版配置比linux下要麻烦一些、且更容易出错,把全过程记录下,以供参考。
1. 准备工作
准备工作与Caffe和TF等框架类似,大部分人可以选择性跳过~
(1)安装CUDA、cuDNN
【CUDA下载地址】https://developer.nvidia.com/cuda-downloads
【cuDNN下载地址】https://developer.nvidia.com/cudnn
本文采用的是CUDA8.0+cuDNN7,把cuDNN的lib和include下文件分别复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下的对应文件夹。
(2)安装Python
caffe2支持Python 2.7.6 - Python 2.7.14。
推荐Anaconda安装,一步到位:
【Anaconda下载地址】https://www.anaconda.com/download/#windows
(3)安装Visual Studio
本文采用的是Visual Studio 2015,VS2017也可以。
【Visual Studio 下载地址】https://www.visualstudio.com/zh-hans/downloads/
(4)安装Cmake
【Cmake下载地址】https://cmake.org/download/
(5)安装依赖库
添加python路径到环境变量中,如 C:\Python27 和 C:\Python27\scripts
安装必需的python依赖:
pip install future ^
hypothesis ^
numpy ^
protobuf ^
six
非必选的python依赖,用于运行caffe2自带的tutorials:
pip install flask ^
glog ^
graphviz ^