Raspberry Pi 上的 Keras GPU 加速？

Question

我很确定这是一个非常愚蠢的问题，但我无法忘记它。所以我确定您知道可以使用 CUDA 或 ROCm 来加速 TensorFlow/Keras 中的学习，但我只是想知道 Raspberry Pi 4 及其 GPU 是否可以帮助训练？

Answer 1

我不知道你所说的“帮助”是什么意思..但在 raspberry pi 4 中，GPU 是一个集成的 Videocore VI。它不支持 CUDA，您不能使用外部 GPU（没有专用于它的连接）。你只能在 CPU 上训练，但是，Raspberry 是一个资源有限的设备，忘掉它吧。您可以在树莓派上进行推理。

Answer 2

您应该在计算机上训练并测试模型。如果有效，请保存您的模型权重和结构并将其部署到 Raspberry Pi。