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

mmclassification 模型转换时出现’tensorrt.tensorrt.Builder’ object has no attribute 'max_workspace_size’问题

tensorrt8.0以上删除了max_workspace_size属性
需要将tensorrt版本降低至7.x

pip install nvidia-tensorrt==7.2.* --index-url https://pypi.ngc.nvidia.com

你可能感兴趣的:(#,TensorRT,mmc,error,tensorrt)