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

以上亲测有效。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐