ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.

傷城~ 2023-09-27 13:36 228阅读 0赞

在复现论文《Mask R-CNN》中，程序用到了PyTorch中的一个包（torch.utils.ffi），但是博主安装的PyTorch版本是1.12.1，也就因此导致了上面的错误。

原因：

在较新版的PyTorch中，torch.utils.ffi被弃用了，需要用其他包来替代。

解决办法：

对于遇到的问题，将原语句：

  1. from torch.utils.ffi import create_extension

修改成:

  1. from torch.utils.cpp_extension import BuildExtension

再将文件下面的调用:

  1. ffi = create_extension( '_ext.nms',
  2.     headers=headers,
  3.     sources=sources,
  4.     define_macros=defines,
  5.     relative_to=__file__,
  6.     with_cuda=with_cuda,
  7.     extra_objects=extra_objects
  8.     )

改成：

  1. ffi = BuildExtension(
  2.     '_ext.nms',
  3.     headers=headers,
  4.     sources=sources,
  5.     define_macros=defines,
  6.     relative_to=__file__,
  7.     with_cuda=with_cuda,
  8.     extra_objects=extra_objects
  9. )

就可以了！

发表评论

表情：
评论列表 （有 0 条评论，228人围观）

还没有评论，来说两句吧...

相关阅读