WIN10+VS2015+PY35+CUDA10.1安装caffe

历时九天,终于成功。

一开始想在Ubuntu上安装,各种BUG,整了整整一星期,放弃。
现在记录下在win10上安装的过程
个人笔记,免得以后安装的时候麻烦。

Caffe版本

caffe用C++编写,要想自己使用,必须进行编译。
在WIN10上总共有三个版本的caffe可以进行编译:

前两种方法首先安装依赖项,修改配置文件,然后利用VS等编译器进行编译。本人用前两种方法均以失败告终,各种BUG,待以后闲了再进行研究吧。Windows版本的修改部分文件,然后运行 最后用的windws版本,直接运行build_win.cmd,成功!

准备工作

安装过程

  • 打开caffe-windows/script/build_win.cmd
    修改以下内容:
    WIN10+VS2015+PY35+CUDA10.1安装caffe

  • 使用CUDA的话要指定CUDNN的路径(此时我已经将cuDNN解压放到CUDA安装目录里了)
    WIN10+VS2015+PY35+CUDA10.1安装caffe

  • 打开caffe-windows\cmake下的Cuda.cmake,作如下更改
    WIN10+VS2015+PY35+CUDA10.1安装caffe
    WIN10+VS2015+PY35+CUDA10.1安装caffe
    这部分不修改会出现下面的错误(原因是显卡性能太高,不在支持系列之内):
    caffe:nvcc fatal : Unsupported gpu architecture 'compute_75’

  • 打开C:\Users<username>.caffe\dependencies\libraries_v140_x64_py27_1.1.0\libraries\include\boost-1_61\boost\config\compiler下的nvcc.hpp,作如下修改:
    WIN10+VS2015+PY35+CUDA10.1安装caffe

安装到python site-package

  • 安装过程会遇到好多警告,大部分好像都是字体的原因(没仔细看),不出错就好。
    WIN10+VS2015+PY35+CUDA10.1安装caffe
  • 将caffe-windows/python文件夹下的caffe直接粘贴到python_root/lib/site-packages/下面,就可以在python环境直接利用caffe了。

遇到的问题

  • 上面一大串,主要是与scikit-img,scipy调用有关的,最后是ImportError: cannot import name ‘Type’。 解决方法,scipy版本不匹配,直接安装了scipy1.0, 出现了其他错误。
  • ImportError: No module named 'numpy.testing.decorators’
    numpy版本不匹配,要安装1.16,我安装的是最新的1.17,部分内容不支持。
  • D:\caffe-windows\python文件夹下的caffe拷贝到site-package中,如下情况,说明安装成功。
    WIN10+VS2015+PY35+CUDA10.1安装caffe

总结

  • 前前后后花了九天时间,不放弃
  • 先是在LINUX下死磕,搞了好长时间,都是各种BUG,以后LINUX还要好好学学。


上面终于完成了caffe的编译,python中import成功,开始进行fast RCNN的调试工作

步骤

  • 将caffe_windows/python下内容全部复制到fast-rcnn\caffe-fast-rcnn下。
  • 由于之前使用python3编译的caffe,这里需要将原始的RBG的python2的代码进行修改。

错误

  • [libprotobuf ERROR C:\Users\guillaume\work\caffe-builder\build_v140_x64\packages\protobuf\protobuf_download-prefix\src\protobuf_download\src\google\protobuf\text_format.cc:298] Error parsing text-format caffe.NetParameter: 195:21: Message type “caffe.LayerParameter” has no field named “roi_pooling_param”.
    F0826 21:48:56.026396 4468 upgrade_proto.cpp:88] Check failed: ReadProtoFromTextFile(param_file, param) Failed to parse NetParameter file: C:\Users\ziliaoshi\Desktop\fast-rcnn\models\CaffeNet\test.prototxt
    *** Check failure stack trace: ***
    原因是我用的不是直接