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

2022-09-02 20:23:14 浏览数 (1)

本文记录了博主遇到问题“ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.”的解决方案。更新于2019.03.12。

背景: 博主需要安装一个程序,该程序用到了PyTorch中的一个包(torch.utils.ffi),但是博主安装的PyTorch版本是1.0.1,也就因此导致了上面的错误。

原因: 在PyTorch 1.0.1中,torch.utils.ffi被弃用了,需要用其他包来替代。

解决办法: 对于博主遇到的问题,将原语句

代码语言:javascript复制
from torch.utils.ffi import create_extension

修改成:

代码语言:javascript复制
from torch.utils.cpp_extension import BuildExtension

再将文件下面的调用:

代码语言:javascript复制
ffi = create_extension(...)

改成:

代码语言:javascript复制
ffi = BuildExtension(...)

0 人点赞