带有 pytorch 的 Android 应用程序

Question

我做了一些安卓应用。我也有 python 文件，它们将一些 jpg 文件作为输入并返回一些值。这些 python 文件使用 pytorch 库。如何在 android 应用程序中运行我的 python 文件？

Answer 1

在 Android 上运行有多种技术，最佳选择取决于您的用例。

如果您的应用程序不需要更快的 GPU 推理速度，Pytorch Mobile 将是最简单的。

如果您需要使用 GPU，则需要将模型转换为在不同的框架上运行。 Tensorflow Lite 在 Android 部署中很受欢迎。如果您主要针对 Qualcomm 设备，则可以通过在 Snapdragon Neural Processing Engine (SNPE) 上运行来提高速度。