在树莓派上使用 gpu 进行 tensorflow 的计算

Question

我正在使用 Tensorflow 在 Raspberry Pi 3 上开发具有计算机视觉功能的机器人。我可以在树莓派 3 上使用 gpu 进行深度学习吗？

Answer 1

更新：

这是 Keras called plaidml 的替代后端，它不是 Tensorflow。主要的卖点是非 Nvidia 显卡的加速。它仍然不是 TensorFlow，但它可能是一个可行的选择。

这是我在 2018-09 之前的旧答案：

简短的回答是否定的，目前不可能，因为 Tensorflow 利用 Nvidia 驱动程序为 Nvidia GPU 提供动力，而 Raspberry Pi 没有 Nvidia 硬件。

您必须改变两件事之一，才能使用 GPU 进行小型计算，Tensorflow 必须支持 OpenCl (tracked here)，或者您必须将平台切换到具有 Nvidia GPU 的平台，例如 this

很抱歉带来坏消息。