如何在GPU上运行网络

时间:2017-09-26 11:30:00

标签: android tensorflow tensorflow-gpu

我开始将我公司的CNN实施集成到我们的Android应用程序中。

作为参考,我使用TensorFlow sample for mobile

不幸的是,我经历了很长的执行时间(大约10-15秒)。在我们的服务器上,只需1秒。

我尝试使用Android Studio中的Android Profiler确定是在CPU还是GPU上执行此操作,并注意到它是在CPU上运行的。

我想与社区分享此问题,以便检查是否有人遇到了与我相同的问题。

我使用的是三星Galaxi A5和Google Pixel。

1 个答案:

答案 0 :(得分:1)

您可能只使用CPU。有两种可能的解释: