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

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

以上亲测有效。

你可能感兴趣的:(TensorRT,深度学习,人工智能)