Я нашел решение. И это был старый добрый совет - «Вы пытались выключить и снова включить?»
Я перезагрузил компьютер, Tensorflow обнаружил cudart64_90.dll
, но теперь он не смог найти cudnn64_7.dll
. Я предоставляю дальнейшие шаги, чтобы избавиться от проблем, с которыми я столкнулся.
Если вы установили версию Tensorflow GPU, вы, скорее всего, столкнетесь с проблемой, упомянутой в посте. Особенно, если вы ранее не устанавливали инструментарий разработки NVDIA. Выполните следующие действия:
1. Установить CUDA
Получите это от здесь . Установите только версию, упомянутую в ImportError Tensorflow.
ImportError: Не удалось найти 'cudart64_90.dll'. TensorFlow требует, чтобы эта DLL была установлена в каталоге, названном в вашей переменной среды% PATH%. Загрузите и установите CUDA 9.0 с этого URL: https://developer.nvidia.com/cuda-toolkit
Он явно сообщает вам номер версии. Первоначально я установил CUDA 9.1 вместо 9.0, он не работал. Установка на Windows прямая. Запустите .exe, снимите флажок NVIDIA Geforce и другие пакеты, если они у вас уже установлены.
2. Включить путь CUDA в переменную PATH
Направьте его в каталог bin вашей установки тензорного потока.
Проверьте здесь , если вы не знаете, как установить переменную PATH. Теперь попробуйте импортировать Tensorflow, если он все еще не работает, попробуйте перезагрузить систему.
Теперь вы, скорее всего, столкнетесь с ошибкой:
ImportError: Не удалось найти 'cudnn64_7.dll'. TensorFlow требует, чтобы эта DLL была установлена в каталоге, названном в вашей переменной среды% PATH%. Обратите внимание, что установка cuDNN - это отдельный шаг от установки CUDA, и эта DLL часто находится в другом каталоге, чем библиотеки CUDA. Вы можете установить необходимую DLL, загрузив cuDNN 7 с этого URL: https://developer.nvidia.com/cudnn
3. Установить cuDNN
Еще раз, установите только версию, указанную в сообщении об ошибке. Чтобы получить программу установки, вам необходимо иметь учетную запись разработчика NVIDIA. Если у вас его нет, зарегистрируйтесь, и он направит вас по ссылке для загрузки cuDNN. Выберите версию, совместимую с вашей версией CUDA (она есть в названии пакета). Загрузите архив zip . Распакуйте его где-нибудь на вашем диске.
4. Включить путь cuDNN в переменную PATH
Аналогично второму шагу. На этот раз укажите его в каталоге bin в извлеченном вами архиве cuDNN. Теперь импортируйте Tensorflow. При необходимости перезагрузите систему.
Теперь должно работать.