AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'max_workspace_size'

[cuge1995]

bugs when convert to TensorRT, my tensorrt version is 8.0.1.6

[shouxieai]

我这里有tensorRT的方案，你可以瞅瞅：https://github.com/shouxieai/tensorRT_cpp

[lijianqing317]

Got the same error when running using TensorRT Python 8.0.0.3. With nvidia-tensorrt-7.2.3.4 it works fine. !pip install nvidia-tensorrt==7.2.* --index-url https://pypi.ngc.nvidia.com https://github.com/NVIDIA-AI-IOT/torch2trt/issues/557#issuecomment-841523481

[sriram-dsl]

builder = trt.Builder(TRT_LOGGER) network = builder.create_network()

builder_config = builder.create_builder_config()

builder_config.max_workspace_size = 1 << 30