В этой статье вы узнаете, как установить и использовать TensorFlow на Ubuntu 18.04 или 16.04. Рассмотрим правильную инсталляцию TensorFlow в виртуальную среду Python или в контейнер Docker. Принципы установки работают в системах Ubuntu и Debian.
Это комплексная платформа с открытым исходным кодом, предназначенная для машинного обучения. Например, TensorFlow может использоваться при построении и обучении нейронных сетей. Благодаря всеобъемлющей и гибкой системе инструментов, библиотек, а также ресурсов сообщества.
TensorFlow позволяет разработчикам использовать самые последние технологии машинного обучения. Теперь стало легче создавать и развертывать приложения на базе машинного обучения. Поисковый гигант Google разработал данную платформу для ранжирования поиска в системе машинного обучения. TensorFlow доступна в виртуальной среде Python или в контейнере Docker. Также можно установить с Anaconda. Для установки вам потребуется уже установленный Python 3.3 и выше, а также Git.
Установка TensorFlow Ubuntu в виртуальную среду Python 3
Шаг 1. Обновление списка пакетов
Убедитесь, что индекс APT по умолчанию обновлен. Для этого выполните следующую команду:
sudo apt update
sudo apt upgrade -y
Шаг 2. Установка Python
Установите Python 3 и Python venv с помощью следующей команды.
sudo apt install python3
sudo apt install python3-pip
sudo apt install python3-venv
После установки Python 3 проверьте, какая версия Python установлена, введя команду:
python3 -V
Шаг 3. Виртуальная среда
Необходимо создать виртуальную среду Python с именем tenorflow_env. Для этого введите следующую команду:
mkdir tensorflow_env
cd tensorflow_env
python3 -m venv my_tensorflow
Эта команда создаст новый каталог my_tensorflow. В нем будут все пакеты, которые вы устанавливаете, пока активирована среда Python.
Теперь нужно активировать виртуальную среду. Иначе мы не сможем ее использовать:
source my_tensorflow/bin/activate
Шаг 4. Установка tensorflow
Осталось просто установить TensorFlow в виртуальной среде, запустив команду инсталляции и обновления до последней версии платформы:
pip3 install --upgrade tensorflow
Для проверки правильности установки и выведения в терминал версии TensorFlow, просто запустите следующую команду:
python -c 'import tensorflow as tf; print(tf.__version__)'
Если все совпадает с изображением на скриншоте, значит вы сделали все правильно.
Установка TensorFlow с помощью Docker
Еще один способ установки TensorFlow – инсталляция в docker-контейнер. Конечно, для начала нужно установить в Ubuntu сам Docker. После этого загрузите файл образа TensorFlow в контейнер Docker, выполнив следующие шаги. Сначала надо загрузить образ TensorFlow для Docker:
sudo docker pull tensorflow/tensorflow
После загрузки образа, запустите его с помощью команды:
docker run -it -p 8888:8888 tensorflow/tensorflow
Больше информации об установке TensorFlow в Docker вы найдете на официальном сайте.
Установка TensorFlow с через pip 3
Существует способ установки с помощью пакетного менеджера pip3. Для этого введите следующую команду:
pip3 install tensorflow
Создание простой программы TensorFlow
После завершения установки проверьте, в каком состоянии находится TensorFlow. Вы можете написать простой «Hello world» код – testTensorFlow.py с помощью текстового редактора vim. Введите команду ниже:
vim testTensorFlow.py
#!/bin/python3
import tensorflow as tf
hello = tf.constant("Hello, world!")
session = tf.Session()
print(session.run(hello))
Далее нужно сохранить и закрыть файл. Затем выполните файл Python следующей командой:
python3 testTensorFlow.py
