почему tf.test.is_gpu_available () не предоставил true или false, он застрял? - PullRequest
1 голос
/ 16 апреля 2020

После установки tenorflow gpu = 2.0.0 он застрял после обнаружения gpu. Настройки окружения для этого проекта:

  1. Ubuntu 18.04
  2. cuda 10.0
  3. cudnn 7.4.1
  4. создано виртуальное окружение
  5. install tenorflow-gpu = 2.0.0

При попытке проверить gpu с помощью tf.test.is_gpu_available (). зависание компиляции показано ниже.

enter описание изображения здесь

1 Ответ

0 голосов
/ 20 апреля 2020

изменена версия cudnn на 7.6.2. Затем она работает хорошо.

...