使用python上的TensorRT.engine文件进行推理

2024-05-28 21:01:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用Nvidia的传输学习工具包(TLT)进行培训,然后使用TLT转换器将.etlt模型转换为.engine文件。在

我想在python文件中使用。但是由于我使用TLT进行了训练,所以我没有任何冻结的图形或pb文件,这正是所有TensorRT推理教程所需要的。在

我想知道在.engine文件上是否可以进行python推断。 如果没有,支持什么样的转换(UFF、ONNX)来实现这一点?在


Tags: 文件模型图形工具包etlt教程enginenvidia

热门问题