第3章代码运行报错：AttributeError: module 'torch' has no attribute 'from_dlpack'

[Ethan-Chen-plus]

运行如下代码：

ex = relax.vm.build(MyModuleWithParams, target="llvm")
vm = relax.VirtualMachine(ex, tvm.cpu())
nd_res = vm["main"](data_nd)
pred_kind = np.argmax(nd_res.numpy(), axis=1)
print("MyModuleWithParams Prediction:", class_names[pred_kind[0]])

报错：

AttributeError                            Traceback (most recent call last)
Input In [31], in <module>
      1 ex = relax.vm.build(MyModuleWithParams, target="llvm")
      2 vm = relax.VirtualMachine(ex, tvm.cpu())
----> 3 nd_res = vm["main"](data_nd)
      4 pred_kind = np.argmax(nd_res.numpy(), axis=1)
      5 print("MyModuleWithParams Prediction:", class_names[pred_kind[0]])

File ~/miniconda3/lib/python3.9/site-packages/tvm/_ffi/_ctypes/packed_func.py:237, in PackedFuncBase.__call__(self, *args)
    225 ret_tcode = ctypes.c_int()
    226 if (
    227     _LIB.TVMFuncCall(
    228         self.handle,
   (...)
    235     != 0
    236 ):
--> 237     raise get_last_ffi_error()
    238 _ = temp_args
    239 _ = args

AttributeError: Traceback (most recent call last):
  4: TVMFuncCall
  3: tvm::runtime::PackedFuncObj::Extractor<tvm::runtime::PackedFuncSubObj<tvm::runtime::relax_vm::VirtualMachine::GetFunction(std::string const&, tvm::runtime::ObjectPtr<tvm::runtime::Object> const&)::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#6}> >::Call(tvm::runtime::PackedFuncObj const*, tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)
  2: tvm::runtime::relax_vm::VirtualMachine::Invoke(long, std::vector<tvm::runtime::TVMRetValue, std::allocator<tvm::runtime::TVMRetValue> > const&)
  1: tvm::runtime::relax_vm::VirtualMachine::RunLoop()
  0: tvm::runtime::PackedFuncObj::Extractor<tvm::runtime::PackedFuncSubObj<TVMFuncCreateFromCFunc::{lambda(tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*)#2}> >::Call(tvm::runtime::PackedFuncObj const*, tvm::runtime::TVMArgs, tvm::runtime::TVMRetValue*) [clone .cold]
  File "/home/kewei/miniconda3/lib/python3.9/site-packages/tvm/_ffi/_ctypes/packed_func.py", line 81, in cfun
    rv = local_pyfunc(*pyargs)
  File "/tmp/ipykernel_6023/484016392.py", line 15, in lnumpy_relu
    x_torch = torch.from_dlpack(x)
AttributeError: module 'torch' has no attribute 'from_dlpack'

[Hzfengsy]

AttributeError: module 'torch' has no attribute 'from_dlpack'

这个是PyTorch的错误，不是特别确定from_dlpack是torch哪个版本引入的。可以考虑将torch升级至最新版本后尝试

[wdhao]

我pytorch 1.9.1都不行，我觉得有可能是版本太高了

[Hzfengsy]

1.9.1版本比较老了，最新版已经1.12了