TensorRT版本:8.4
从网上下载的代码,运行报错,如上述标题。
1、错误:AttributeError: ‘tensorrt.tensorrt.Builder‘ object has no attribute ‘max_workspace_size‘
原因:tensorrt8.0以上删除了max_workspace_size属性。
解决方案:(1)将tensorRT降低为7.x版本(不推荐)
(2)使用如下命令替换:
# builder.max_workspace_size = 1 << 20
config = builder.create_builder_config()
config.max_workspace_size = 1 << 20
以上亲测有效。