1

我正在尝试运行tensorflow-federated在 GPU 上使用的 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
4

1 回答 1

2

很难确切地说出这里的问题是什么，但我确实有一个怀疑。

TFF将 TensorFlow 2.1 声明为必需包；这可能意味着您的 TF-gpu 安装在 TFF 安装设置期间被删除。

请您尝试以下顺序：

  • 开个新鲜venv
  • 安装 TFF
  • 卸载 TF
  • 安装 TF GPU 2.1.0
于 2020-04-12T16:58:53.407 回答