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

Question

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

Answer 1

如果您想使用 Rasperry Pi在主流框架（例如 TensorFlow 或 PyTorch）内训练神经网络，您可能在短期内无法获得 GPU 加速。原因是至少这两个都是以 GPU 为中心的主流，主要是 NVidia，逐渐启用 AMD 支持（这里是 TF，这里是 Torch）。

可以说，Pi 最有前途的选择是OpenCL 支持，但是，截至撰写本文时，据我所知，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 模型的转换工具，但在实现的功能方面相当有限。