最新的Windows Caffe配置方法(Python接口)

首先,重新隆重介绍Windows Caffe的下载地址:

https://github.com/BVLC/caffe/tree/windows

真是非常感谢大牛们不辞劳苦地维护和更新这些代码,为我们的科研提供了无限可能。

有一些预编译好的程序,我们有需要可以直接去官网下载。这里说一下自主配置和编译的过程。


1 需要的软件

Visual Studio 2013 或者 2015

CMake 3.4 或者更高版本(支持VisualStudio 和 Ninja)

 

2 可选的依赖项

Python 用于pycaffe接口。AnacondaPython 2.7 或者3.5 x64(或者Miniconda)

MATLAB用于matcaffe 接口

CUDA 7.5 或者 8.0(如果使用VS2015,请使用CUDA 8.0)

cuDNN v5

 

3 配置和生成Caffe

我们先要把cmake.exe和python.exe加到PATH环境变量上。

假设我们要把Caffe装到C:\Projects路径下,那么我们可以打开Windows的命令行窗口,cd到该路径下,执行下面的命令

C:\Projects> git clonehttps://github.com/BVLC/caffe.git

C:\Projects> cd caffe

C:\Projects\caffe> git checkout windows

:: Edit any of the options insidebuild_win.cmd to suit your needs

C:\Projects\caffe> scripts\build_win.cmd

 

上面用到的build_win.cmd脚本会从网上下载依赖文件,创建VS工程(或者ninga生成文件),并且以Release的配置生成整个工程。默认情况下,需要的DLLs会被拷贝(或者采用硬链接方式)到二进制文件旁。

 

4 在生成过程中需要做的一些工作

我采用的配置是开启GPU、python接口。GPU是默认开启的,Python接口需要做一些简要配置。

如果使用了Python的发行版Anaconda或者Miniconda,使用下面的命令来安装一些Python需要的package。

condainstall --yes numpy scipy matplotlib scikit-image pip six

还需要一个与生成好的pre-built依赖兼容的protobuf的Python package。

conda install --yes --channel willyd protobuf==3.1.0

 

如果在上述的caffe的编译过程中遇到依赖项下载失败问题,可以考虑自己先单独从指定网址下载依赖库文件到本地,放到指定目录。然后再重新编译。在caffe\scripts目录下有download_prebuilt_dependencies.py脚本,打开它可以看到下载依赖项的网址。



posted @ 2017-03-27 15:01  雁回晴空  阅读(228)  评论(0编辑  收藏  举报