由于为个人安装的是TensorRT10
,有些参数可能就不用了,在模型转换的时候会出现错误(本文章长期更新)
特别注意onnxruntime_gpu
的安装,他是与CUDA和CUDANN版本
进行绑定的
安装时候参考网址
错误1
YOLOv8模型转换的时候,出现了
attributeerror: 'tensorrt.tensorrt.builder' object has no attribute 'max_workspace_size'
这个是由于TensoRT8.4版本之后,取消了这个max_workspace_size
,这个可以这样修改
1注销掉
#config.max_workspace_size = self.args.workspace * 1 << 30
2、紧接其后,添加这个
config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, 1<<30) # fix TRT 8.4 deprecation notice
3、write file的修改为这个
with builder.build_serialized_network(network, config) as engine, open(f, 'wb') as t:
4、写文件修改为:
#t.write(engine.serialize())
t.write(engine)