python - 如何在 Python 中导入 caffe 模块？

Question

我在 Windows（版本，x64）上构建了 _caffe.cpp 的 .dll。

我将扩展名 .dll 更改为 .pyd 并尝试在 python 中导入它：

import caffe

  File "\caffe-master\python\caffe\__init__.py", line 1, in <module>
    from .pycaffe import Net, SGDSolver

  File "\caffe-master\python\caffe\pycaffe.py", line 13, in <module>
    from ._caffe import Net, SGDSolver

ImportError: DLL load failed: The specified module could not be found.

这是什么意思，缺少一些依赖项模块，这些模块包含在 Visual Studio 的项目中，我在其中构建了这个 dll？

Answer 1

您需要将 Python Caffe 添加到 PYTHONPATH。例如： 导出 PYTHONPATH=$PYTHONPATH:/home/username/caffe/python

Answer 2

对于窗户：

添加/caffe/Build/x64/Release/pycaffe到系统路径（path）对我有用，我认为最好的方法是：

1. 新建一个系统变量：PYTHON_PKG = /caffe/Build/x64/Release/pycaffe;
2. 包括PYTHON_PKG在path：path = %PYTHON_PKG%; %OtherDirs%

在我这样做之后，我丢失了 PKG google.internal，然后我pip install google.internal在CMD. 有用。

Answer 3

一旦你有一个编译和构建的 caffe，尝试

echo 'export PYTHONPATH=/path/to/caff-dir/python'

此外，您可能需要运行以下命令：

pip install -r requirement.txt