TensorRT文档建议
Created by: OliverLPH
-
版本、环境信息 1)PaddlePaddle版本:Paddle1.5 2)CPU/GPU:CUDA8,cuDNN7 , P4 3)系统环境:Ubuntu1604
-
建议描述:
ISSUE10129 就像这个issue里提到的,官方的TensorRT lib,在编进Paddle的时候需要手动添加 ~IPluginFactory()。
所以我觉得应该可以在官网提供Paddle适配好的TRT的lib吧,这样感觉对用户也比较友好。
或者就像在Docker file里注明的一样,在文档里注明也可以。
- 复现信息:
还有就是在 CUDA8,cuDNN7 编译从Nvidia官网下的 TensorRT-4.0.1.6.Ubuntu-16.04.4.x86_64-gnu.cuda-8.0.cudnn7.1.tar.gz 这个包,不光class IPluginFactory 需添加 ~IPluginFactory()。class IGpuAllocator 也需要添加 virtual ~IGpuAllocator() {} 才能编得过。