如何解决“不推荐使用torch.utils.ffi。请改用cpp扩展”而不降级pytorch版本？

Question

当我运行下面的代码时，它会显示错误。 ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead. 我一直在网上寻找解决方案。问题是下面的代码在旧版本的火炬（0.4.1）中工作。我想知道是否可以修改或替换此代码以在新版本的 pytorch 中工作。

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())

Answer 1

我也面临同样的问题，刚刚在以下网站看到了一些有用的信息：

• https://pytorch.org/tutorials/advanced/cpp_extension.html
• https://pytorch.org/docs/stable/cpp_extension.html

为避免降级 PyTorch 的版本，您应该考虑使用以下库，同时在上述链接中查找更多详细信息：

from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CppExtension