使用CMake为Caffe编译Python接口

Question

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

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

有没有人有这方面的经验？有没有人通过使用cmake编译python接口？

感谢。

Answer 1

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