В моей первой попытке Tensorflow я использовал контейнер Docker для Tensorflow, доступный на сайте. Итак, я подумал об установке Tensorflow на свой компьютер с Windows 10.
Изменить: Если вы хотите связаться со мной, не стесняйтесь писать мне по адресу teavanist [at] gmail [dot] com; Средний не очень способствует разговору
Оглавление
- Предпосылки
- Установить Python 3.6
- Установить Tensorflow GPU 1.5
- Установить CUDA Toolkit 9.0
- Скопируйте cudnn64_7.dll из библиотеки cuDNN
- Добавить папку CUDA в ПУТЬ
- Проверить установку Tensorflow
Предварительные условия
Вам необходимо учитывать совместимость CUDA и Tensorflow. Я использовал следующие версии:
Python - 3.6.6
Tensorflow GPU - 1.5.0
CUDA для Windows 10 - 9.0
CUDNN для Windows 10 - 9.0
Я полагался на следующее руководство, без которого я бы потратил несколько дней на устранение ошибок, с которыми я столкнулся изначально:
Моя машина с Windows 10 оснащена графическим процессором NVIDIA GeForce GT 610.
NVIDIA опубликовала на своем сайте список графических процессоров с поддержкой CUDA:
Установите Python 3.6
Изначально я установил последнюю версию python и сразу же приступил к установке Tensorflow. Но я не мог продолжить, потому что столкнулся с множеством ошибок. Немного погуглив, я понял, что текущая версия Tensorflow не поддерживается в последней версии Python, то есть версии 3.7 (которую я установил).
Поэтому я удалил Python 3.7 и загрузил версию 3.6.6 с сайта Python.
Я выбрал 64-битный исполняемый файл
Установка довольно проста. Дважды щелкните программу установки и следуйте инструкциям на экране после включения следующих параметров:
Откройте командную строку в Windows и проверьте установку, как показано на следующем рисунке:
Установите Tensorflow GPU 1.5
Теперь откройте командную строку и введите следующую команду:
pip install tensorflow-gpu==1.5.0
После завершения установки вы увидите сообщение, подобное этому:
Установите CUDA Toolkit 9.0
Загрузите версию CUDA Toolkit 9.0 по следующей ссылке:
После загрузки дважды щелкните, чтобы начать установку.
Во время установки вы увидите следующие сообщения:
После подписания лицензионного соглашения убедитесь, что вы выбрали вариант «Выборочная установка», снимите флажок Интеграция с Visual Studio и нажмите «Далее».
Установка должна завершиться, и теперь нам нужно извлечь
Скопируйте cudnn64_7.dll из библиотеки cuDNN
Для работы Tensorflow требуется файл cudnn64_7.dll. Этот файл доступен в zip-архиве в архиве cuDNN:
После загрузки библиотеки откройте zip-файл.
Перейдите в cuda- ›bin, и вы увидите файл cudnn64_7.dll:
Этот файл необходимо добавить в папку bin в папке CUDA. На моем ПК программа была установлена в следующем месте:
C: \ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v9.0 \ bin
Распакуйте файл dll в папку bin.
Добавить папку CUDA в ПУТЬ
Нам нужно отредактировать переменные среды с местоположением CUDA. Вы можете просто ввести редактировать системные переменные среды в строке поиска Windows, и появится опция:
Когда откроется окно свойств системы, щелкните «Переменные среды».
В следующем окне выберите Путь и нажмите Изменить:
Затем нажмите «Создать», вставьте местоположение папки bin и нажмите «ОК».
Проверьте установку Tensorflow
Вы можете проверить установку Tensorflow, используя следующие команды в командной строке Windows:
> python > import tensorflow as tf > print (tf.__version__)
Вы обнаружите, что версия Tensorflow 1.5.0 будет отображаться, как показано ниже: