无法在GPU上运行Tensorflow-联合

Question

我正试着在图形处理器上运行我的python代码，它使用tensorflow-federated。要设置我的环境，我使用venv。首先，我安装了tensorflow-gpu，然后我的python代码可以识别图形处理器，我使用tf.test.gpu_device_name()。然而，一旦我安装了tensorflow-fedenerated，我的python就不会看到任何图形处理器，而是开始使用CPU！我使用的是Ubuntu 16.04.6 LTS。我尝试了大量不同版本的包的组合：

python = 3.6, 3.7
cuda = 10.0, 10.1
tensorflow-gpu = 1.13.1, 1.15, 2.1.0, 2.0.0-alpha0
tensorflow-fedenerated = 0.2.0, 0.12.0, 0.13.0

Answer 1

很难说这里到底是什么问题，但我确实有点怀疑。

TFF declares TensorFlow 2.1 as a required package；这可能意味着您的TF-gpu安装在TFF安装设置过程中被删除。

你能试试下面的顺序吗？

• 打开全新的venv
• Install TFF
• 卸载TF
• 安装TF GPU 2.1.0