CAFFE环境配置教程

CAFFE 环境配置教程(win10+vs2015)

说明:本教程适用于在windoows平台下搭建caffe环境,搭建环境前请确认自己的电脑系统为win10,有N卡GPU。

1.安装VS2015

在微软官网上下载vs2015的社区免费版并安装,之前安装过其他版本的VS不用卸载。

2.下载安装CUDA8.0

根据自己的电脑配置选择相应的下载包,尽量选择下载本地安装包,在线安装包可能会出现小错误。

3.下载安装CUDNN v5.0

注意这里的CUDNN版本,若下载其他版本的CUDNN可能与CUDA8.0不兼容
下载cuDNN v5(官网下载相应的系统版本),将解压出的三个文件夹剪切到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0覆盖相应文件夹。
cudnn下载

4.下载caffe的第三方支持包thirdparty
将百度云中的thirdparty压缩包解压到 caffe-windows-ms\windows\thirdparty并将caffe-windows-ms\windows\thirdparty\bins的完整路径添加到path环境变量。

5.将mnist.zip解压到路径caffe-windows-ms\data\mnist\
批处理文件create_mnist-lmdb.bat 和 train_lenet.bat 复制到
caffe-windows-ms\examples\mnist

6.将caffe-windows-ms/windows/中的文件CommonSettings.props.example重命名为CommonSettings.props,打开GPU Compute Capability查找对应的显卡计算容量, 3.0的需要将CommonSettings.props(用写字板打开)中的

<CudaArchitecture>compute_52,sm_52;compute_60,sm_60</CudaArchitecture>
    改为
<CudaArchitecture>compute_30,sm_30;compute_52,sm_52</CudaArchitecture>
其他显卡改成
    <CudaArchitecture>compute_35,sm_35;compute_52,sm_52</CudaArchitecture>

7.需要python支持的往下拉找python支持,建议尽量安装这个。
(笔者当时没有安装Python相关配置,导致LIBCAFFE中的一个Python相关的文件编译失败)
确认CommonSettings.props中部分配置如下。
配置
修改

<MXNetSupport>false</MXNetSupport>

修改props配置文件中的CudnnPath和PythonDir
8.用VS打开caffe-windows-ms\windows\caffe.sln,设置release模式。
若提示需要加载相关组件如Python组件,则按提示操作

9.对libcaffe项目右键生成,生成成功后再生成caffe.sln的解决方案

解决方案

10.测试环境是否配置成功
caffe-windows-ms\examples\mnist\create_mnist-lmdb.bat
制作数据,train_lenet.bat训练数据,可以编辑更改
(结果日志应该在C:\Users\用户名\AppData\Local\Temp内以“caffe.exe.”开头)
训练截图如下。
这里写图片描述

PYTHON支持安装

  1. 安装anaconda python 27版本,官网免费下载安装。
  2. 进入CMD运行下面代码
    conda install protobuf
    如安装过其他Python可以使用
    pip install protobuf
  3. 添加环境变量PythonDir指向anaconda安装目录
    修改.\windows\CommonSettings.props 中的python 路径
    <PythonDir> C:\Users\xxxx\Anaconda2\</PythonDir>
    指向anaconda安装目录。
  4. 继续安装教程第8步
  5. 安装完成后添加环境变量,“PythonPath” 指向 caffe_root\Build\x64\Release\pycaffe
  6. 进入cmd依次输入python ,import caffe测试是否通过
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值