美文网首页
win10下的caffe环境配置

win10下的caffe环境配置

作者: 阿Way的无人之境 | 来源:发表于2017-04-26 19:52 被阅读0次

    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测试是否通过

    相关文章

      网友评论

          本文标题:win10下的caffe环境配置

          本文链接:https://www.haomeiwen.com/subject/sjjwzttx.html