无法在GPU上运行Tensorflow联合

Question

我正在尝试在GPU上运行使用tensorflow-federated的python代码。要设置我的环境，我使用venv。首先，我安装tensorflow-gpu，然后我的python代码可以识别GPU，我使用tf.test.gpu_device_name()。但是，一旦我安装了tensorflow-fedenerated，我的python就不再看到任何GPU并开始使用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;这可能意味着在安装TFF的过程中，删除了您的TF-gpu。

可以请您尝试以下命令：

• 打开新的Remote Method Call
• 安装TFF
• 卸载TF
• 安装TF GPU 2.1.0