Caffe编译(Win10+Vs2015+Cuda8.0)

本文详细介绍了在Windows 10上使用Visual Studio 2015和CUDA 8.0编译Caffe的步骤,包括环境配置、编译配置修改、依赖包下载和Caffe的编译过程。此外,还提供了测试MNIST数据集的流程,包括数据准备、prototxt文件修改和执行训练与测试。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参考博客


  1. 编译
  2. 测试minist数据集

环境


  1. Win10
  2. Vs2015 or Vs2013
  3. cmake-3.7.2-win64-x64.zip
  4. cudnn-8.0-windows10-x64-v5.1
  5. Cuda8.0
  6. Anaconda2-4.3.0.1-Windows-x86_64 (Python2.7 or 3.5)
  7. Git-2.12.0-64-bit
  8. Caffe-Windows源码

编译步骤


  1. 安装上述所有环境
  2. 配置环境变量
    • cmake的bin目录加入path
    • cudnn的cuda下bin目录加入path
    • Anaconda安装目录加入path(python就在根目录下)
  3. 解压Caffe-Windows,最好不要含有中文路径
  4. 进入…\caffe-windows\scripts,用文本编辑器如Notepad++打开build_win.cmd文件(我的目录F:\vs2015\caffe\caffe-windows\scripts\build_win.cmd),下面进行修改编译配置:

          我的配置如下,如果你是Vs2013,把MSVC_VERSION=14改为12。WITH_NINJA=0,除非你想用ninja编译器来进行编译。Python版本设置为2或者3,取决于装的版本,这里编译只支持2.7或者3.5版本,其他不用怎么改变。

    else (
        :: Change the settings here to match your setup
        :: Change MSVC_VERSION to 12 to use VS 2013
        if NOT DEFINED MSVC_VERSION set MSVC_VERSION=14
        :: Change to 1 to use Ninja generator (builds much faster)
        if NOT DEFINED WITH_NINJA set WITH_NINJA=0
        :: Change to 1 to build caffe without CUDA support
        if NOT DEFINED CPU_ONLY set CPU_ONLY=0
        :: Change to Debug to build Debug. This is only relevant for the Ninja generator the Visual Studio generat
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值