В PyCharm:
Вариант 1. Если вы установили TensorFlow без поддержки графического процессора, вам не нужно ничего делать. Просто импортируйте TensorFlow в ваш скрипт, и все готово.
Вариант 2: Если вы установили TensorFlow с поддержкой GPU, вы также должны установить CUDA Toolkit 9.0 и cuDNN v7.0 (24.05.2018). Вы можете добавить соответствующие каталоги в переменную окружения% PATH%. Однако даже тогда PyCharm жаловался! Поэтому гораздо проще добавить эти каталоги непосредственно в структуру проекта PyCharm.
Файл> Настройка> Структура проекта> Добавить корень содержимого> Добавить
· path_to_Cuda \ v9.0 \ bin
· path_to_cuDNN \ v7.0 \ bin
Импортируйте TensorFlow в ваш скрипт, и все готово:)
Надеюсь, это поможет!