win10 vs2015 编译caffe2

安装vs2015 cuda8 python3 cmake

   

    

     

      

     
     

      

       git clone --recursive https://github.com/caffe2/caffe2.git
      
     
    

     

      

     
     

      

     
   
  

   

    

     

      

     
     

      

       用记事本打开CAFFE2_ROOT/scripts/build_windows.bat
      
     
    

     

      

     
     

      

       在文件首的 @echo off之下添加
      
     
    

     

      

     
     

      
   set USE_CUDA=ON  （如果只是使用CPU，不使用GPU，那么这行就不要添加）
      
     
    

     

      

     
     

      

       将   -DBUILD_PYTHON=OFF^  改为   -DBUILD_PYTHON=ON^
      
     
    

     

      

     
     

      

       之后就可以执行 build_windows.bat，成功之后会出现提示信息： "Caffe2 built successfully"
      
     
    

     

      

     
     

      

       在CAFFE2_ROOT目录下会出现 build文件夹
      
     
   
  

 CMake Error at third_party/protobuf/cmake/cmake_install.cmake:39 (file):
27>    file cannot create directory: C:/Program Files/Caffe2/lib.  Maybe need
27>    administrative privileges.
27>  Call Stack (most recent call first):

27>    cmake_install.cmake:37 (include)

13. 安装Caffe2

1.先编译proto

  

   

    

     

    
    

     

      右击 INSTALL 项目，build
     
    
   

    

     

    
    

     

      完成之后会在 C:\Program Files 出现 Caffe2 文件夹
     
    
   

    

     

    
    

     

      将 CAFFE2_ROOT\build\caffe2\python\Release 下的 
     
    
   

    

     

    
    

     

      caffe2_pybind11_state.pyd 和 caffe2_pybind11_state_gpu.pyd 拷贝到 C:\Python27\DLLs
     
    
   

    

     

    
    

     

      配置环境变量： PYTHONPATH = CAFFE2_ROOT\build 或者
     
    
   

    

     

    
    

     

      PYTHONPATH = C:\Program Files\caffe2
     
    
  
 

• 1
• 2
• 3
• 4
• 5
• 6

14. 测试caffe2，跑mnist

在任意目录新建 test.py，代码如下

 

  

   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import shutil
     
    
   

    

     

    
    

     

      from caffe2.python import core, cnn, net_drawer, workspace, visualize
     
    
   

    

     

    
    

     

      core.GlobalInit(['caffe2', '--caffe2_log_level=0'])
     
    
   

    

     

    
    

     

      caffe2_root = "~/caffe2"
     
    
   

    

     

    
    

     

      print("Necessities imported!")
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/79935481