如何使用 GPU 测试 TensorFlow Federated？

Question

我正在尝试在 Anaconda Python 中设置虚拟环境，但我不断遇到依赖问题。请告诉我哪个版本的tensorflow_federated与tensorflow-gpu 2.5.0兼容。虚拟环境中的Python版本是3.8

pip安装tensorflow-gpu==2.5.0 pip 安装tensorflow_federated

我尝试通过网络搜索更改tf-gpu和tff的版本，但我仍然遇到同样的问题。

Answer 1

我正在尝试在 Anaconda Python 中设置虚拟环境

请按照链接设置虚拟环境。

请告诉我哪个版本的tensorflow_federated与tensorflow-gpu 2.5.0兼容。虚拟环境中的Python版本是3.8

Tensorflow-federated 与 3.9 之前的 Python 版本不兼容。 您需要更新虚拟环境中的python版本。