torch.utils.ffi 已弃用。如何改用 cpp 扩展？

Question

我正在尝试运行使用来自 torch.utils.ffi 的 wrap_function 的代码（已弃用）。我正在努力弄清楚如何按照错误消息的建议使用 cpp 扩展，有人可以帮忙吗？

我需要替换的代码：

from torch.utils.ffi import _wrap_function
from ._nms import lib as _lib, ffi as _ffi

__all__ = []
def _import_symbols(locals):
    for symbol in dir(_lib):
        fn = getattr(_lib, symbol)
        if callable(fn):
            locals[symbol] = _wrap_function(fn, _ffi)
        else:
            locals[symbol] = fn
        __all__.append(symbol)

_import_symbols(locals())

我已尝试在 python 3.6 和 python 2.7 中运行代码，请参阅下面的错误消息，

  File "build.py", line 3, in <module>
    from torch.utils.ffi import create_extension
  File "/usr/local/lib/python2.7/dist-packages/torch/utils/ffi/__init__.py", line 1, in <module>
    raise ImportError("torch.utils.ffi is deprecated. Please use cpp extensions instead.")
ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.

Answer 1

检查

from torch.utils.cpp import ....
#or
from torch.utils.cpp_extension import ....
#or
from torch.utils.cpp_extension_versioner import...

在我的电脑上看起来不错