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

本文记录了博主遇到问题“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被弃用了，需要用其他包来替代。

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

from torch.utils.ffi import create_extension

 

修改成:

from torch.utils.cpp_extension import BuildExtension

 

再将文件下面的调用:

ffi = create_extension(...)

 

改成:

ffi = BuildExtension(...)

 

就可以了。

---------------------  
作者：ShuqiaoS  
来源：优快云  
原文：https://blog.youkuaiyun.com/ShuqiaoS/article/details/88420326  
版权声明：本文为博主原创文章，转载请附上博文链接！