caffe python3.5 Windows版本编译

不念不忘少年蓝@ 2022-03-09 03:24 249阅读 0赞

之前编译的是caffe python2.7版本，与caffe python3.5版本还是有差异，记录一下。

下载 caffe windows版本：[https://github.com/BVLC/caffe/tree/windows][https_github.com_BVLC_caffe_tree_windows]

前提：已安装 anaconda 3（python 3.5），VS2015，CMake，CUDA8.0+cudnn5.1

**一 caffe-windows\\scripts路径下修改配置文件build\_win.cmd**

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70][]

如上图， 根据提示VS、Python版本修改为相应的版本号(MSVC\_VERSION=14：VS2015；PYTHON\_VERSION=3：Python3.5)；CPU\_ONLY=0：GPU。

**二** 修改好build\_win.cmd文件，可以打开**CMake进行配置生成caffe的VS项目了**（也可直接输入命令.\\scripts\\build\_win.cmd）

CMake中，如图1所示，source code位置为caffe路径，build为新建的生成文件路径，下图红圈圈处BLAS其Value由Atlas改为Open，否则会出现“Could NOT find Atlas...”的错误。点击Configure，选择相应的编译器（此处VS2015），CMake会下载相应的依赖包libraries\_v140\_x64\_py35\_1.1.0.tar.bz2到C:\\Users\\XX\\.caffe\\dependencies\\download下，编译时会到dependencies找相应的包（例如BLAS会调用openblas的包），配置成功，再点击Generate，完成。得到caffe的解决方案Caffe.sln。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 1][]

图1  CMake配置节目

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 2][]

图2   编译器选择

可能出现的问题：本地存在python多个版本，比如其他应用安装了python 2.7，configure时先识别python 2.7之后，会在dependencies自动下载对应python 2.7版本的依赖包libraries\_v120\_x64\_py27\_1.1.0.tar.bz2，可以先屏蔽安装python 2.7版本的应用。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 3][]

图3   configure界面

**三 VS编译caffe**

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 4][]

以ALL\_BUILD为启动项，配置：Release+x64，编译整个工程，完成。

![20190312153153530.png][]

可以在 caffe-windows\\build\\tools\\Release 下调用相应的可执行文件了

![20190312153443696.png][]

**四 python中调用caffe**

Caffe.sln中pycaffe编译好之后，在caffe-windows\\python目录下会生成caffe文件夹，将其拷贝到Anaconda3\\Lib\\site-packages目录下。此时在Spyder中调用应能成功调用caffe。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 5][]

然而：import caffe 时出现了 from numpy.lib.arraypad import \_validate\_lengths， ImportError: cannot import name '\_validate\_lengths'的错误，应该是scikit-image包的问题，对其更新 pip install --upgrade scikit-image之后，又出现caffe cannot import name 'img\_as\_float32'，直接pip uninstall 再install scikit-image，此时import caffe成功。

[https_github.com_BVLC_caffe_tree_windows]: https://github.com/BVLC/caffe/tree/windows
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70]: /images/20220309/71de6b5e35f24f4fbfecc55e543d551f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 1]: /images/20220309/e9db42737f3848c6964d95e483bbe0de.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 2]: /images/20220309/652f7619cc3a4774834fa1943d88c6ef.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 3]: /images/20220309/ba92c6c575a8445aa014ca424a2ddc2a.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 4]: /images/20220309/cc39129b8f214bc6992bff016fb866a9.png
[20190312153153530.png]: /images/20220309/cc01a177b8ba4cf7aedc4aa79cb4d58f.png
[20190312153443696.png]: /images/20220309/b769aee20b67413fbdc627ac9bef1120.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2doeV8xMTE_size_16_color_FFFFFF_t_70 5]: /images/20220309/d0cbe6e7e0744a0fb66620864bef2ec9.png