最新的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脚本,打开它可以看到下载依赖项的网址。



转载于:https://www.cnblogs.com/yanhuiqingkong/p/7770054.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值