caffe-windows的配置过程

来源:互联网 发布:淘宝达人头像图片 编辑:程序博客网 时间:2024/06/06 09:41
这里讲使用GPU和不使用GPU的配置过程:

配置准备过程:
   1. 从官网下载Caffe:https://github.com/Microsoft/caffe
   2. 若使用GPU加速则需从官网下载CUDA和cuDnn,不使用GPU的则无需下载
       CUDA下载地址:https://developer.nvidia.com/cuda-downloads
       cuDnn下载地址:https://developer.nvidia.com/cudnn
      使用GPU加速的则对PC有要求,需英伟达的显卡才可加速。否则使用CPU,虽然速度慢些,但不影响效果。
   3.安装VS

一.不使用GPU的配置过程
     1. 进入caffe-master\windows路径下,复制CommonSettings.props.example做备份。
     2. 重新命名CommonSettings.props.example为CommonSettings.props
     3. 打开CommonSettings.props,修改7.8行内容如下:
        <CpuOnlyBuild>true</CpuOnlyBuild>
        <UseCuDNN>false</UseCuDNN>
       保存并关闭CommonSettings.props
    4.双击caffe.sln,默认用VS打开,release下单击菜单生成 -> 重新生成解决方案,开始漫长的编译过程
    
     在预编译阶段,VS会通过NuGet工具自动获取预编译的Caffe依赖包,放在caffe-master同一个文件夹的NugetPackages下。

二. 使用GPU的配置过程
      1. 安装CUDA和下载cuDnn压缩包
      2. 进入caffe-master\windows路径下,复制CommonSettings.props.example做备份。
      3. 重新命名CommonSettings.props.example为CommonSettings.props
      4. 打开CommonSettings.props,修改内容如下:
         <CpuOnlyBuild>false</CpuOnlyBuild>
         <UseCuDNN>true</UseCuDNN>
         <CudaVersion>8.0</CudaVersion>  改成自己的版本
         <CuDnnPath>D:\caffe\caffe-master-GPU\cudnn-8.0-windows10-x64-v5.1</CuDnnPath> 改成自己的路径
        
         保存并关闭CommonSettings.props
 
      5. 双击caffe.sln,默认用VS打开,点击libcaffe,右键->属性,点击配置属性- >c/c++ 将警告视为错误改为No,release下单击菜单生成 -> 重新生成解决方案,开始漫长的编译过程
    
         在预编译阶段,VS会通过NuGet工具自动获取预编译的Caffe依赖包,放在caffe-master同一个文件夹的NugetPackages下。
          
        
参考微博:http://www.cnblogs.com/love6tao/p/5706830.html
           



0 0
原创粉丝点击