【已解決】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(...)

就可以了。