GPU跑tensorflow

[复制链接] · 发表于 2019-2-25 16:17:42

本帖最后由 vlsida2019 于 2019-2-25 16:18 编辑

如何使用开发板上的GPU 跑tensorflow模型如何使用开发板上的GPU 跑tensorflow模型

如何使用开发板上的GPU 跑tensorflow模型

只看该作者 · 发表于 2019-2-25 17:40:13

参考tensorflow_lite教程就可以了，这个google官方android8.1支持的。
但是lite支持的op可是相当的少的。

只看该作者 · 发表于 2019-2-25 18:17:52

jefferyzhang 发表于 2019-2-25 17:40
参考tensorflow_lite教程就可以了，这个google官方android8.1支持的。
但是lite支持的op可是相当的少的。 ...

比NPU的还少吗

只看该作者 · 发表于 2019-2-26 10:32:54

程子发表于 2019-2-25 18:17
比NPU的还少吗

你试试就知道了，比npu少的不是一点半点。
目前支持的op数量是 TF > RKNN > TFLITE
他们当时移植时候考虑过直接支持tflite，后来发现lite的op少的简直像是要重写一遍，工作量巨大