BerryLJ

you can try from torch.utils.ffi import create_extension change from torch.utils.cpp_extension import BuildExtension，your code will be work!