Не удается запустить Tensorflow-federated на графическом процессоре

Question

Я пытаюсь запустить мой код python, который использует tensorflow-federated на графическом процессоре. Чтобы настроить свою среду, я использую venv. Сначала я устанавливаю tensorflow-gpu, и мой код python затем может распознать графический процессор, я использую tf.test.gpu_device_name(). Однако, как только я установлю tensorflow-fedenerated, мой python перестанет видеть любой графический процессор и начнет использовать процессоры !! Я использую 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 объявляет TensorFlow 2.1 как необходимый пакет ; это может означать, что ваша установка TF-GPU была удалена во время установки TFF.

Можете ли вы попробовать следующий порядок:

• Открыть файл sh venv
• Установить TFF
• Удалить TF
• Установить TF GPU 2.1.0