win10 x64环境下配置caffe(包含python3.5接口)

版权声明:本博主原创文章,欢迎转载。 https://blog.csdn.net/qq_24915933/article/details/88879446

按照caffe 官方在windows环境下的要求配置caffe,生成pycaffe库

我在这里将 详细介绍如何使用 vs2015 + CMake + Ninja 的方式在win10 x64环境下配置python3.5的caffe第三方可供调用的pycaffe包。 在这里插入图片描述

CMake 安装与配置

CMake官网下载安装程序,但注意版本要>=3.4。直接安装CMake,但在安装的时候要记得勾选将CMake包含在电脑的环境变量中;如果安装时没有勾选,那也没关系,直接找到CMake的安装路径,将其bin文件所在路径添加到环境变量中,如下图:
在这里插入图片描述

VS2015 community版 安装

直接在microsoft官网下载安装2015版本的社区版即可。

Git 安装

Git直接在官网下载win64下面的安装包直接安装即可,Git用来下载Ninja和caffe到本地电脑上。

Ninja安装与配置

Ninja的安装依赖于vs2015(或者2013),因此一定要安装好vs2015之后再来安装Ninjia。Ninja安装步骤如下:

  • 打开cmd命令窗口,在某个路径下,下载Ninja包,如在C:\wq_documents\ZTE\caffe 路径下, 输入以下命令,将会联网将Ninja clone到该目录下面,然后会进入ninja子目录:
git clone git://github.com/ninja-build/ninja.git && cd ninja

在这里插入图片描述

  • 在ninja下载完成后,以管理员的身份打开vs2015 x64 Native Tools Command(一定要是此窗口,而不能直接打开cmd窗口进行ninja编译),如下图:
    在这里插入图片描述
  • 打开该 vs2015 x64 native tools command命令窗口 后,定位到ninja所在的路径,输入下面的命令,将会编译ninja,直到完成。但是在这一步中,一定要将python添加到环境变量中,不然会报错(如果是用conda安装的虚拟python(我使用的是python3.5)环境,也可以将python.exe所在的路径添加到环境变量中,像添加CMake到环境变量中一样):
python ./configure.py --bootstrap

在这里插入图片描述

  • 编译完成后,会出现下列东西在ninja路径当中,同时,一定要将此路径添加到环境变量中,不然到时候用CMake编译caffe会报错
    在这里插入图片描述
    将ninja添加到环境变量中:
    在这里插入图片描述
    到这一步,我们已经准备好了编译caffe的所有前提准备了,一定要确保所有的环境变量都配置好了。 后面将介绍如何编译caffe。

caffe 编译

  • 修改 scripts目录下 build_win.cmd 文件,可以让WITH_NINJA=1,表示用Ninja编译(或者直接利用vs2015 直接编译也可以, 不用安装Ninja)
    在这里插入图片描述
  • 利用Git clone caffe-windows到本地文件夹(路径随便你设置),输入以下命令即可(但是 在执行build_win.cmd之前, 一定要将build_win.cmd文件中的python版本按要求该为3,不然会出现python版本不匹配的问题 ),等待caffe编译完成,就可以了:
git clone https://github.com/BVLC/caffe.git 
cd caffe
git checkout windows
scripts\build_win.cmd

在这里插入图片描述

  • 在anaconda创建的虚拟环境下,输入一下命令,是能够使用pycaffe重要的步骤:
conda config --add channels conda-forge
conda config --add channels willyd
conda install --yes cmake ninja numpy scipy protobuf==3.1.0 six scikit-image pyyaml pydotplus graphviz

在这里插入图片描述

  • 将编译好的caffe包拷贝至虚拟环境的site-packages中,如下图所示:

在这里插入图片描述
编译好的caffe python接口有以下内容
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_24915933/article/details/88879446