search

NVIDIA-AI-IOT / torch2trt

An easy to use PyTorch to TensorRT converter
MIT License
4.56k stars 673 forks source link

AttributeError: 'tensorrt_bindings.tensorrt.INetworkDefinition' object has no attribute 'add_convolution'. Did you mean: 'add_convolution_nd'? #920

Open AWallyAllah opened 5 months ago

AWallyAllah commented 5 months ago

I'm converting a PyTorch model with MobileNetV2 backbone, However I get this error:

AttributeError: 'tensorrt_bindings.tensorrt.INetworkDefinition' object has no attribute 'add_convolution'. Did you mean: 'add_convolution_nd'?

Versions:

xmdszzz commented 5 months ago

@AWallyAllah hey bro, i checked the latest release notes, v0.4.0 only supports up to TensorRT 8, so TensorRT10 is not supported, and you should change the TRT's version.

jaybdub commented 4 months ago

Hi @AWallyAllah , @xmdszzz,

We've just added support for TensorRT 10 in the master branch.

Could you try this?

Let me know if you run into any issues or this works for you.

Best, John

XXMY commented 1 month ago

Hi @AWallyAllah , @xmdszzz,

We've just added support for TensorRT 10 in the master branch.

Could you try this?

Let me know if you run into any issues or this works for you.

Best, John

running succeed!