如何在 Python 中導入 caffe 模塊？

Question

我已經在 Windows (Release, 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。 例如： export PYTHONPATH=$PYTHONPATH:/home/username/caffe/python

Answer 2

對於窗戶：

將/caffe/Build/x64/Release/pycaffe到系統路徑（ path ）對我/caffe/Build/x64/Release/pycaffe ，我認為最好的方法是：

1. 新建一個系統變量： PYTHON_PKG = /caffe/Build/x64/Release/pycaffe;
2. 在path包含PYTHON_PKG ： path = %PYTHON_PKG%; %OtherDirs% path = %PYTHON_PKG%; %OtherDirs%

完成此操作后，我得到 PKG 缺少google.internal ，然后我在CMD了pip install google.internal 。 有用。

Answer 3

編譯並構建了 caffe 后，請嘗試

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

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

pip install -r requirement.txt