nvinfer1::cudnn::Engine::deserialize()出错的原因

  根据工作要求,特征提取更改了模型。同事传了一个TensorRT engine文件。一加载,出错如下:

python3.5: cudnnEngine.cpp:710: bool nvinfer1::cudnn::Engine::deserialize(const void*, std::size_t, nvinfer1::IGpuAllocator&, nvinfer1::IPluginFactory*): Assertion `size >= bsize && "Mismatch between allocated memory size and expected size of serialized engine."' failed.

  人们不禁要问……表问了,这跟人性扭曲、道德沦丧一点关系也没有。赶紧想办法解决。

  首先,网上搜索了一下,Stackoverflow说有一个类似的错误,没有人回答。这个是一个同事转换的模型,于是问这位高手,把源码、模型都发吾。然后一对比,发现体积相差太大,试了一下就成功了。

你可能感兴趣的:(CUDA/TensorRT)