python - 有没有办法在 Raspberry Pi 4 上获得 GPU 加速以进行深度学习？

Question

在官方 Tensorflow Lite 网站上，他们展示了对 Android 和 iOS 的 GPU 加速的支持，但他们没有说明对 Raspberry Pi 的支持。

Answer 1

如果您想将 Rasperry Pi 用于 培训 TensorFlow 或 PyTorch 等主流框架内的神经网络，您可能无法在短期内获得 GPU 加速。原因是至少这两个是以主流 GPU 为中心的，主要是 NVidia，并且逐渐启用 AMD 支持(here 用于 TF，here 用于 Torch)。
可以说，Pi 最有前途的选择是 OpenCL support但是，在撰写本文时，据我所知，Raspberry Pi 4 还没有 OpenCL 实现。
可能与 TensorFlow.js你可以得到 Raspberry Pi GPU 的一些东西，只要它可以通过 WebGL 着色器使用 GPU。因此，在 Pi 上运行的具有 WebGL 支持的网络浏览器可以启用 GPU 加速训练。
如果您想将 Raspberry Pi 用于 推理 ，可能更容易受益于 GPU 加速。

ncnn是一个使用 Vulkan API 为 Raspberry Pi 4 启用 GPU 加速的库。

Beatmup库支持对所有 Raspberry Pi 模型进行 GPU 加速推理，不仅仅是 4。它还提供了一些来自 Keras 模型的转换工具，但在实现的功能方面受到很大限制。