Установка TensorFlow

DIGITS теперь поддерживает TensorFlow в качестве дополнительного альтернативного бэкенда для Caffe или Torch.

Установка для Ubuntu (https://www.tensorflow.org/install/install_linux#installing_with_virtualenv)

Установка для Mac (https://www.tensorflow.org/install/install_mac#installing_with_virtualenv)

Требования

В настоящее время DIGITS ориентирован на TensorFlow-GPU версии 1.2.1.

Для работы с TensorFlow для DIGITS требуется один или несколько графических процессоров NVIDIA с вычислительной мощностью CUDA 3.0 или выше. См. официальный список поддержки GPU, чтобы узнать, поддерживает ли его ваш графический процессор.

Наряду с этим требованием должны быть установлены:

• Один или несколько графических процессоров NVIDIA (подробности)
• Драйвер NVIDIA (подробная информация и инструкции по установке)
• Набор инструментов CUDA (подробная информация и инструкции по установке)
• cuDNN 5.1 (страница загрузки)

Примечание о cuDNN и TensorFlow

В настоящее время TensorFlow версии 1.2.1 ориентирован на cuDNN версии 5.1. Чтобы использовать TensorFlow версии 1.2.1 в DIGITS, необходимо установить cuDNN версии 5.1. Для этого используйте следующую команду в терминале:

sudo apt-get install libcudnn5

Установка

Эти инструкции основаны на [официальных инструкциях TensorFlow] (https://www.tensorflow.org/versions/master/install/).

TensorFlow поставляется с pip, чтобы установить его, просто используйте команду:

pip install tensorflow-gpu==1.2.1

Затем TensorFlow должен легко установиться и подтянуть все необходимые зависимости.

Начало работы с TensorFlow в DIGITS

Следуйте этим инструкциям для получения информации о начале работы с TensorFlow в DIGITS.