转自:https://github.com/NVIDIA-AI-IOT/torch2trt/issues/557
有两种解决方法:
1.将tensorrt降级回7版本;
2.使用如下命令替换:
# builder.max_workspace_size = 1 << 20
config = builder.create_builder_config()
config.max_workspace_size = 1 << 20
转自:https://github.com/NVIDIA-AI-IOT/torch2trt/issues/557
有两种解决方法:
1.将tensorrt降级回7版本;
2.使用如下命令替换:
# builder.max_workspace_size = 1 << 20
config = builder.create_builder_config()
config.max_workspace_size = 1 << 20