tensorRT8 save/load Egine

 save:

	    std::ofstream p(filename, std::ios::binary);
		auto gieModelStream = mEngine->serialize();
		p.write((const char*)gieModelStream->data(), gieModelStream->size());
		p.close();
		gieModelStream->destroy();

load :

    struct stat my_stat;
	char* filename = "./onnx_engine.engine";

	if (stat(filename, &my_stat) == 0) {

		initLibNvInferPlugins(&gLogger.getTRTLogger(), "");

		std::ifstream engineFile(filename, std::ios::binary);
		if (!engineFile)
		{
			std::cout << "Error opening engine file: " << filename << std::endl;
			return nullptr;
		}

		engineFile.seekg(0, engineFile.end);
		long int fsize = engineFile.tellg();
		engineFile.seekg(0, engineFile.beg);

		std::vector<char> engineData(fsize);
		engineFile.read(engineData.data(), fsize);

		SampleUniquePtr<IRuntime> runtime{ createInferRuntime(gLogger.getTRTLogger()) };

		auto mEngine = runtime->deserializeCudaEngine(engineData.data(), engineData.size(), nullptr);

		nvinfer1::Dims mInputDims = mEngine->getBindingDimensions(0);
     }

 亲测good! tensorRT8;

**************************************************************************************************************

#include "NvInferPlugin.h"

initLibNvInferPlugins(&gLogger.getTRTLogger(), ""); 

不加这个会报错!!!!!!!

错误代码:[pluginV2Runner.cpp::nvinfer1::rt::load::290] Error Code 1: Serialization (Serialization assertion creator failed.Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry)
[12/14/2021-10:38:14] [E] [TRT] 4: [runtime.cpp::nvinfer1::Runtime::deserializeCudaEngine::50] Error Code 4: Internal Error (Engine deserialization failed.)

参考:Developer Guide :: NVIDIA Deep Learning TensorRT Documentation

 TensorRt5: How to save the engine after it has built - #5 by yuvaramsingh94 - TensorRT - NVIDIA Developer Forums

How to serialize TensorRT model into file and deserialize from file? - #2 by piotrw - TensorRT - NVIDIA Developer Forums

  • 1
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 10
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值