tensorflow predict 的 C++接口比 Python 速度还慢?

2019-01-08 08:40:03 +08:00
 zhiqiang

tensorflow 提供了 C++和 Python 接口,测试了两个接口的速度,结果发现 C++比 Python 还慢。

为什么呢,有人来说说不?

我们只测试了 predict。模型是事先训练好的。在计算运行时间时,只统计两边调用 predict 函数的时间。

3796 次点击
所在节点    问与答
6 条回复
luxinxin
2019-01-08 08:44:20 +08:00
冷启动时间是否排除掉了,以及 cpp 和 python 接口是否用的是同样的 batch size
zhiqiang
2019-01-08 08:58:28 +08:00
@luxinxin 你之前有类似的测试吗?

我只统计了 tensorflow 的 predict 函数调用时间,tensorflow 启动和数据准备时间都被排除了。除非在 predict 函数内部的启动,这个也无法排除。

batch size 一样。模型也一样。数据也一样。
luxinxin
2019-01-08 09:15:55 +08:00
我没做过类似的对比,要不你再试试先 predict 一千个样本,这样能把 predict 内部的启动因素也排除掉,然后再开始计时。
zhiqiang
2019-01-08 09:30:54 +08:00
@luxinxin 我做了十万次然后对比的时间。。。
glasslion
2019-01-08 10:16:03 +08:00
慢了多少? C++ 的 tensorflow 和 Python 的 tensorflow 是否完全一致(mkl 等)?
zhiqiang
2019-01-09 09:38:58 +08:00
@glasslion 不多,一半到一倍吧。

编译一致性这个我不清楚,怎么查?

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/524838

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX