windows10下基于vs2015的 caffe安装教程及python接口实现 - 善良的大猪猪

windows10下基于vs2015的 caffe安装教程及python接口实现

啦啦啦:根据网上的教程前一天安装失败,第二天才安装成功。其实caffe的安装并不难,只是网上的教程不是很全面,自己写一个,留作纪念。

准备工作

Windows10 操作系统

vs2015(c++编译器一定要已经安装)

cmake编译工具(加入到环境变量)

anaconda2(我第一次用的anaconda3,用的最新的版本安装失败,结果发现caffe支持的是python3.5的anaconda,所以选择了python2.7也就是anaconda2)

下载caffe

直接下载Windows版的就行(官方提供的)

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

1. 下载.zip解压快一点,不用下载git软件还要git clone xxx 速度慢。并放到某一目录下我的是E:\Program Files\

2 .在script中有一个download_prebuilt_dependencies.py,我的是E:\Program Files\caffe-windows\scripts,直接下载第二个。

WIN_DEPENDENCIES_URLS = {
    (\'v120\', \'2.7\'):("https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v120_x64_py27_1.1.0.tar.bz2",
                  "ba833d86d19b162a04d68b09b06df5e0dad947d4"),
    (\'v140\', \'2.7\'):("https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py27_1.1.0.tar.bz2",
                  "17eecb095bd3b0774a87a38624a77ce35e497cd2"),
    (\'v140\', \'3.5\'):("https://github.com/willyd/caffe-builder/releases/download/v1.1.0/libraries_v140_x64_py35_1.1.0.tar.bz2",
                  "f060403fd1a7448d866d27c0e5b7dced39c0a607"),
}

  如果用的是anaconda3(基于python3.5)就选择第三个下载。等会有用。

3. 更改‪E:\Program Files\caffe-windows\scripts\build_win.cmd的内容

    if NOT DEFINED CPU_ONLY set CPU_ONLY=1  #我的电脑没有GPU
    if NOT DEFINED CUDA_ARCH_NAME set CUDA_ARCH_NAME=Auto
    if NOT DEFINED CMAKE_CONFIG set CMAKE_CONFIG=Release
    if NOT DEFINED USE_NCCL set USE_NCCL=0
    if NOT DEFINED CMAKE_BUILD_SHARED_LIBS set CMAKE_BUILD_SHARED_LIBS=0
    if NOT DEFINED PYTHON_VERSION set PYTHON_VERSION=2 #PYTHON版本为python2
    if NOT DEFINED BUILD_PYTHON set BUILD_PYTHON=1   #加入python接口,也就是想用 import caffe 这个操作
    if NOT DEFINED BUILD_PYTHON_LAYER set BUILD_PYTHON_LAYER=1
    if NOT DEFINED BUILD_MATLAB set BUILD_MATLAB=0
    if NOT DEFINED PYTHON_EXE set PYTHON_EXE=python

  将下面的代码复制到else里面或者下面(我加到了103-110行)

if !PYTHON_VERSION! EQU 2 (
        set CONDA_ROOT=E:\Program Files\conda2 #自己根据情况更改
    )
    :: Set python 3.5 with conda as the default python
if !PYTHON_VERSION! EQU 3 (
        set CONDA_ROOT=E:\Program Files\conda2  #这里可以不看,代码运行不到这里
    )
set PATH=!CONDA_ROOT!;!CONDA_ROOT!\Scripts;!CONDA_ROOT!\Library\bin;!PATH!

4. 直接运行‪E:\Program Files\caffe-windows\scripts\build_win.cmd。最好在命令行里运行,出错看得到日志。

运行完成没有错误的情况下会生成build文件夹,有警告没有关系。有错误的话根据错误修改代码重新执行(先删除build文件夹)

5 build 文件夹中会出现caffe.sln, 别打开,等一下还有工作没有做。

6 将第二步下载的解压,在caffe目录下新建build文件夹,将解压结果labraries文件夹放到build文件夹下,我的是E:\Program Files\caffe-windows\build\libraries

并将

E:\Program Files\caffe-windows\build\libraries\bin

E:\Program Files\caffe-windows\build\libraries\lib

E:\Program Files\caffe-windows\build\libraries\x64\vc14\bin

加入到环境变量(很重要,不然就缺少东西)

7. 可以打开caffe.sln了,直接生成解决方案即可。有警告没有错误不用管

8. 将E:\Program Files\caffe-windows\python\caffe加入到python的库里。

我加到了E:\Program Files\conda2\Lib\site-packages

基本完成,哈哈哈

执行conda2的python,import caffe

也就这样咯

PS E:\Program Files\conda2> .\python.exe
Python 2.7.15 |Anaconda, Inc.| (default, May  1 2018, 18:37:09) [MSC v.1500 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
>>>

  

发表于 2018-06-25 17:24 善良的大猪猪 阅读(454) 评论(0) 编辑收藏举报