Tensorflow-federated 0.17.0 完全是 GPU 还是 GPU 和 CPU？

【问题标题】：Tensorflow-federated 0.17.0 is totally GPU or GPU and CPU?Tensorflow-federated 0.17.0 完全是 GPU 还是 GPU 和 CPU？
【发布时间】：2020-11-30 09:53:40
【问题描述】：

我想知道新版本的 TFF (0.17.0) 是否完全是 GPU ？因为在旧版本上，我发现 TFF 运行 CPU 和 GPU。

【问题讨论】：

标签： tensorflow-federated

【解决方案1】：

TFF 0.17.0 旨在在 GPU 上运行（如果可用），否则根据底层 TensorFlow pip 包使用 CPU。这与https://www.tensorflow.org/guide/gpu 内联：

如果 TensorFlow 操作同时具有 CPU 和 GPU 实现，则默认情况下，在将操作分配给设备时会优先考虑 GPU 设备。

如果不是这种情况，请在http://github.com/tensorflow/federated 上打开一个问题，并包含pip freeze 的输出、有关环境的信息，包括操作系统、python 版本和硬件（系统中存在哪些 GPU）。

【讨论】：