2017-09-26 134 views
0

我開始將我公司的CNN實現集成到我們的Android應用程序中。作爲參考,我使用TensorFlow sample for mobile如何讓GPU在網絡上運行

不幸的是,我正在經歷很長的執行時間(大約10-15秒)。在我們的服務器上,只需要1秒。

我試着找出這個執行是在使用Android Studio中的Android Profiler發生在CPU還是GPU上,並且發現它在CPU上運行。

我想與社區分享這個問題,以檢查是否有人遇到與我一樣的問題。

我正在使用Samsung Galaxi A5和Google Pixel。

回答

1

您可能只使用CPU。有兩種可能的解釋:

+0

能否請您與我分享支持的GPU的鏈接?我可以找到一個。關於你分享的鏈接,它也會告訴你,當你使用Makefile來構建靜態庫時,它不支持GPU。我非常肯定,這對於Android的Gradle構建系統來說是一樣的。 Gradle也構建.so,可能不支持GPU。我想知道該解決方案是否會構建我自己的.so庫,與TensorFlow服務交流,並嘗試自己啓用GPU功能。 – Nativ

+0

更新:我找到了下面的鏈接,指定了TensorFlow for iOS的限制。其中之一是GPU不受支持。我相信這也涵蓋了Android,因爲TensorFlow Mobile核心是跨平臺的。 http://machinethink.net/blog/tensorflow-on-ios/ – Nativ

+0

「您能否與我分享支持GPU的鏈接?」我沒有一個。但是我可以看到有關使用NVidia GPU的TensorFlow的頁面。 –