如何在统一移动设备中使用 tensorflow-gpu答案

【问题标题】：How to use tensorflow-gpu in unity mobile如何在统一移动设备中使用 tensorflow-gpu
【发布时间】：2019-11-28 06:09:15
【问题描述】：

我正在 Unity 引擎中对移动机器学习应用程序进行原型设计。
我已经训练了张量流图（.pb），我想在统一移动设备中运行模型。（安卓和ios）
使用 OpenCVForUnity 插件和 dnn 模块，我可以在移动设备中运行 tensorflow 图。但问题是它在 CPU 上运行。

我需要基于 GPU 的解决方案，而 OpenCVForUnity 似乎不适合。
那么在统一移动环境中在 GPU 上运行图形有什么想法吗？

【问题讨论】：

【解决方案1】：

您可能想要使用Barracuda，它可以让您转换 tensorflow 模型并在跨平台 Unity 应用程序中使用它。 Unity ML-Agents 使用梭子鱼，因此您可以使用他们的代码作为如何利用您的神经网络的参考。

【讨论】：