【发布时间】:2017-06-14 05:22:40
【问题描述】:
我的最终目标是使用 TensorRT 来优化我的部署模型。我正在张量流中做我的实验。但是 TensorRT 需要 caffemodel 格式的输入模型。
【问题讨论】:
标签: tensorflow deep-learning nvidia tensorflow-gpu tensorrt
我的最终目标是使用 TensorRT 来优化我的部署模型。我正在张量流中做我的实验。但是 TensorRT 需要 caffemodel 格式的输入模型。
【问题讨论】:
标签: tensorflow deep-learning nvidia tensorflow-gpu tensorrt
截至 TensorRT 2.1,不支持转换 TensorFlow 模型。将来可能会支持此功能。
您可能想要探索将 TensorFlow 模型转换为 Caffe 的选项,然后使用 TensorRT 将其转换为推理引擎。
【讨论】:
新的 TensorRT 3 添加了对 TensorFlow 的支持。
【讨论】: