使用 CMake 为 Caffe 编译 Python 接口答案

【问题标题】：Compiling the Python Interface for Caffe using CMake使用 CMake 为 Caffe 编译 Python 接口
【发布时间】：2015-10-11 18:41:16
【问题描述】：

在 Ubuntu 上，我已将 Caffe 下载到 ~/caffe，并使用 cmake 和它附带的 CMakeLists.txt 文件对其进行编译。我现在想开始使用 python 接口。根据示例，我需要将~/caffe/python 添加到我的.bashrc 文件中的PYTHONPATH 变量中。然后，我需要在 python 脚本中使用 import caffe 。但是，这给了我错误：ImportError: No module named 'caffe'。

如果我在我的 python 脚本中使用import caffe 行，这意味着在PYTHONPATH 定义的目录中必须存在一个名为caffe.py 的文件。那是对的吗？但是，我在我的系统上找不到任何名为 caffe.py 的文件。但是CMakeLists.txt 文件应该用于构建python 文件，并且有add_subdirectory(python) 行。

有人有这方面的经验吗？有没有人通过 cmake 编译让 python 接口工作？

谢谢。

【问题讨论】：

I have downloaded Caffe to ~/caffe, and compiled it using cmake - 你有没有在构建后安装 caffe？

标签： python cmake makefile caffe

【解决方案1】：

~/caffe/python 中不需要有 caffe.py。在这种情况下，您应该有一个子文件夹~/caffe/python/caffe，该子文件夹中应该有__init__.py 文件。确保您有 ~caffe/python/caffe/_caffe.so 文件 - 此共享对象应在编译期间创建。
可能是您在.bashrc 中添加了正确的文件夹，但您是否在更新后加载/获取了该文件？

【讨论】：