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

时间：2021-10-26 21:43:32

啦啦啦：根据网上的教程前一天安装失败，第二天才安装成功。其实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

>>>

　　

相关文章