zoukankan      html  css  js  c++  java
  • Win10上使用VS2015编译Caffe2

    Caffe2的官网:https://caffe2.ai/

    1、下载、安装及相关准备

    在Caffe2的官网点击"Get Started",即进入安装说明页面。官方还未提供编译好的binaries,所以需要自己使用源代码编译。

    选择平台如下:

    按照官网,需要Win10系统及以上,但是似乎有些博客也有使用了Win7,但我本人是使用Win10,所以本文只针对Win10系统展开。

    在准备阶段,我主要下载或安装的有:

    1、Python2.7.14 https://www.python.org/downloads/release/python-2714/ 

    2、cmake 3.11 https://cmake.org/download/

    3、CUDA9.0 https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

    4、cuDNN v7.0.5 for CUDA9.0 https://developer.nvidia.com/rdp/cudnn-archive

    5、visual stduio 2015 Enterprise(用Community和Professional应该也是没问题的,只是刚好有人给我Enterprise的安装包)

    6、Python的库,只安装了必须的几个:future、hypothesis、numpy、protobuf、six,可选的那些我都没安装(我是使用pip install安装的 )

    7、另外,我原来自己还有安装了Opencv3.4,因为我是使用Caffe2来处理图像的

    8、安装git,版本应该无所谓吧,主要是用来下载源码

    9、下载caffe2源码, caffe2(https://github.com/caffe2/caffe2)现在已和pytorch(https://github.com/pytorch/pytorch)合并了,所以使用git下载:

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

    接着说一下安装过程中的一些注意或者遇到的问题:

    1、安装vs2015一般选自定义,然后勾选Visual C++的相关选项。

    2、安装vs失败很麻烦,因为有时候卸载不干净会影响CUDA安装里面Visual studio integrate组件的安装(稍后会写另一篇博客来说明)

    3、下载caffe2源码不要直接Download Zip下载压缩包,那样会导致第三方库缺失,即third_party文件夹里面的一些文件。但是目前从github上之间git clone源码速度很慢,都是10+k甚至几k/s,更崩溃的是下载途中还会经常中断,然后又得重新来了,所以我试了很多次都没成功,FQ速度也很慢(也可能是我的VPN不给力),后来是托人在香港那边下载的。

    4、下载cuDNN解压缩后,是一个包含include、lib和bin的文件夹,将这三个文件夹里的文件分别拷贝到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0文件夹下的include、lib和bin下(当然,以你的CUDA的安装路径为准,我是采用默认安装路径

    2、编译安装

    git得到pytorch文件夹后,在pytorch的scripts文件夹里有build_windows.bat脚本文件,可以用记事本、Notepad或者sublime text等编辑器打开,如果懂得脚本和Cmake命令就可以完全读懂。里面主要需要做的是指定编译器和开启编译选项。在该文件里加入这两句,可以在“pip install pyyaml”语句上方:

    set USE_CUDA=ON
    set CMAKE_GENERATOR="Visual Studio 14 2015 Win64"

    CMAKE_GENERATOR即是设置编译器,USE_CUDA只是是否使用CUDA,由于我是要使用GPU,如果你不需要则不必加。

    另外,下面的cmake命令则可以设置编译选项,比如是否使用GLOG、GFLAGS等等第三方库,由于我本人是要使用Caffe2来部署训练好的模型,所以我只开启了USE_OPENCV,如下图所示。

    接着以管理员身份打开vs2015开发人员命令提示,即Developer Command Prompt。使用cd命令至pytorch的script文件夹下,然后运行build_windows.bat,编译需要稍长的时间。

    编译成功后,在pytorch文件夹下的build文件夹里,使用vs打开Caffe2.sln。我选择了Release x64平台下,右键ALL_BUILD生成解决方案。然后再右键INSTALL生成解决方案,这两个过程也是比较久的。INSTALL编译成功后,在C:Program Files就出现了Caffe2文件夹,里面主要存放了生成的include、lib和bin。

  • 相关阅读:
    python note 30 断点续传
    python note 29 线程创建
    python note 28 socketserver
    python note 27 粘包
    python note 26 socket
    python note 25 约束
    Sed 用法
    python note 24 反射
    python note 23 组合
    python note 22 面向对象成员
  • 原文地址:https://www.cnblogs.com/jisongxie/p/9435755.html
Copyright © 2011-2022 走看看