如何在 Python 中导入 caffe 模块

How to import caffe module in Python?

本文关键字：导入 caffe 模块 Python      更新时间：2023-10-16

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

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

import caffe
  File "caffe-masterpythoncaffe__init__.py", line 1, in <module>
    from .pycaffe import Net, SGDSolver
  File "caffe-masterpythoncaffepycaffe.py", line 13, in <module>
    from ._caffe import Net, SGDSolver
ImportError: DLL load failed: The specified module could not be found.

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

你需要将 Python Caffe 添加到 PYTHONPATH 中。例如：export PYTHONPATH=$PYTHONPATH：/home/username/caffe/python

对于窗口：

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

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

在我这样做之后，我得到了PKG丢失google.internal，然后我在CMD中做了pip install google.internal。它有效。

一旦你有一个编译和构建的咖啡，试试

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

此外，您可能需要运行以下内容：

pip install -r requirement.txt
相关文章：