Caffe学习笔记01——基于Windows安装Caffe(CPU版)

本文详细指导Windows环境下安装Caffe库,包括环境配置、Visual Studio选择、CMake设置、Python版本调整及依赖包管理。从修改build_win.cmd脚本开始,通过编译并验证Mnist示例,确保Caffe成功搭建。
摘要由CSDN通过智能技术生成

参考的官方配置文档:GitHub - BVLC/caffe-windows

1、环境准备

环境清单:

  • 操作系统:Windows10
  • Windows版Caffe源码
  • Visual Studio 2015
  • CMake
  • Python3.5
  • 依赖包

详细操作:
①Windows版Caffe源码:官网下载,下载解压后,文件名是caffe-windows,如果解压完名字是caffe则说明下载错了。

②Visual Studio 2015:可以在 下载之家 或者 软件学堂 进行下载,下载community版即可,下载ultimate版还需要破解。安装时组件里面一定要选上Visual C++,因为后面编译时,会用到里面的C/C++编译器,不选的话编译时会报错:

-- The C compiler identification is unknown
-- The CXX compiler identification is unknown

③CMake:下载2016年以后的任意版本都行,我是在网上下载解压后直接用的,需要将bin目录放到系统环境变量下。CMake是编译时,会用到的一个工具,如果没安装好的话,编译时会报错:

'cmake' 不是内部或外部命令,也不是可运行的程序或批处理文件。

④Python3.5:如果版本不对的话,编译时会报错:

Could not find url for MSVC version = 1900 and Python version = 3.8

安装一个Anaconda可以方便地切换环境,在Anaconda里新建一个Python3.5的环境,然后将对应的路径放到环境变量path的最上面(Python的路径有多个时,系统默认用最上面的那个),实在不行的话就下载Python3.5版本的Anaconda。

⑤依赖包:编译时要下载很多的依赖包,下载速度慢而且容易报错,所以我们提前下载好,下载地址
使用:先进行编译,软件会自动创建对应的文件夹并进行下载,我们Ctrl+C取消编译,然后将下载的依赖包放到这个文件夹中。我的默认地址是:

C:\Users\hp\.caffe\dependencies\download

2、编译

caffe-windows代码解压后,找到scripts文件夹下的build_win.cmd文件,其中build_win.cmd的作用是下载依赖项、创建VS项目文件、构建版本配置。
①修改build_win.cmd,主要是69行-100行之间。

else (
    :: Change the settings here to match your setup
    :: Change MSVC_VERSION to 12 to use VS 2013
    ::  根据自己vs版本设置,2013就是12,2015就是14,我的是14
    if NOT DEFINED MSVC_VERSION set MSVC_VERSION=14 
    :: Change to 1 to use Ninja generator (builds much faster)
    :: 这里由于我使用的cmake编译,未使用ninja,所有设置为了0
    if NOT DEFINED WITH_NINJA set WITH_NINJA=0
    :: Change to 1 to build caffe without CUDA support
    :: 这里我们安装的是CPU版本所以设置为1;
    if NOT DEFINED CPU_ONLY set CPU_ONLY=1
    :: Change to generate CUDA code for one of the following GPU architectures
    :: [Fermi  Kepler  Maxwell  Pascal  All]
    if NOT DEFINED CUDA_ARCH_NAME set CUDA_ARCH_NAME=Auto
    :: Change to Debug to build Debug. This is only relevant for the Ninja generator the Visual Studio generator will generate both Debug and Release configs
    :: 此处就是编译vs的版本,是Release还是Debug版本,我的是Release,Release优化了代码训练速度更快。
    if NOT DEFINED CMAKE_CONFIG set CMAKE_CONFIG=Release
    :: Set to 1 to use NCCL
    :: 是否使用多卡训练
    if NOT DEFINED USE_NCCL set USE_NCCL=0
    :: Change to 1 to build a caffe.dll
    if NOT DEFINED CMAKE_BUILD_SHARED_LIBS set CMAKE_BUILD_SHARED_LIBS=0
    :: Change to 3 if using python 3.5 (only 2.7 and 3.5 are supported)
    :: 选择python版本,我使用是python 3.5,所以修改为3
    if NOT DEFINED PYTHON_VERSION set PYTHON_VERSION=3
    :: Change these options for your needs.
    :: 这里选择需要的选项,我需要python,所以将python选项设置为了1,如果需要MATLAB,则将其也设置为1
    if NOT DEFINED BUILD_PYTHON set BUILD_PYTHON=1
    if NOT DEFINED BUILD_PYTHON_LAYER set BUILD_PYTHON_LAYER=1
    if NOT DEFINED BUILD_MATLAB set BUILD_MATLAB=0
    :: If python is on your path leave this alone
    if NOT DEFINED PYTHON_EXE set PYTHON_EXE=python
    :: Run the tests
    if NOT DEFINED RUN_TESTS set RUN_TESTS=0
    :: Run lint
    if NOT DEFINED RUN_LINT set RUN_LINT=0
    :: Build the install target
    :: 生成安装文件,所以需要设置为1
    if NOT DEFINED RUN_INSTALL set RUN_INSTALL=1
)

②修改好build_win.cmd文件后保存退出,然后在caffe-windows文件下打开命令行工具,输入:

.\script\build_win.cmd

之所以在cmd里面运行是因为出错了会显示提示信息,直接运行build_win.cmd出错了就直接退出了,cmd的一些基本操作:

打开窗口:WIN+R后输入cmd
切换到G盘:G:
切换到指定目录:cd G:\caffe-windows\scripts
执行文件:build_win.cmd
注意:改变环境变量后cmd窗口要关了重开

等待编译结束:
编译结束

③生成解决方案(对整个项目进行编译):

G:\caffe-windows\scripts\build

这个路径下有个Caffe.sln文件,用Visual Studio打开并且生成解决方案
生成解决方案
最后全部成功,没有失败即可
生成成功

3、验证

①scripts文件下会多出一个build文件
②G:\caffe\caffe-windows\scripts\build\tools\Release里面有很多编译生成的可执行文件
③可以做一个Mnist的实验,来验证一下。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值