Как установить TensorFlow в Ubuntu 20.04

TensorFlow — это бесплатная платформа с открытым исходным кодом для машинного обучения, созданная Google. Он используется рядом организаций, включая Twitter, PayPal, Intel, Lenovo и Airbus.

TensorFlow можно установить в масштабе всей системы, в виртуальной среде Python, как контейнер Docker или с Anaconda.

В этом руководстве объясняется, как установить TensorFlow в виртуальной среде Python на Ubuntu 20.04.

Виртуальная среда позволяет вам иметь несколько разных изолированных сред Python на одном компьютере и устанавливать определенную версию модуля для каждого проекта, не беспокоясь о том, что это повлияет на другие ваши проекты.

Установка TensorFlow в Ubuntu 20.04

Ubuntu 20.04 по умолчанию поставляется с Python 3.8. Вы можете убедиться, что Python 3 установлен в вашей системе, набрав:

python3 -V

Результат должен выглядеть так:

Python 3.8.2

Рекомендуемый способ создания виртуальной среды — использование модуля venv , который входит в пакет python3-venv .

Чтобы установить пакет python3-venv , выполните следующую команду:

sudo apt install python3-venv python3-dev

После установки модуля вы готовы создать виртуальную среду для нашего проекта TensorFlow.

Перейдите в каталог, в котором вы хотите сохранить виртуальные среды Python 3. Это может быть ваш домашний каталог или любой другой каталог, в котором ваш пользователь имеет права на чтение и запись.

Создайте новый каталог для проекта TensorFlow и перейдите в него:

mkdir my_tensorflowcd my_tensorflow

В каталоге выполните следующую команду, чтобы создать виртуальную среду:

python3 -m venv venv
Второй venv — это имя виртуальной среды. Вы можете использовать любое имя для виртуальной среды.

Приведенная выше команда создает каталог с именем venv , который содержит копию двоичного venv Python, диспетчер пакетов Pip , стандартную библиотеку Python и другие вспомогательные файлы.

Чтобы начать использовать виртуальную среду, активируйте ее, запустив сценарий activate :

source venv/bin/activate

После активации каталог bin виртуальной среды будет добавлен в начало системной $PATH . Кроме того, приглашение оболочки изменится, и в нем будет отображаться имя виртуальной среды, в которой вы находитесь. В этом примере это (venv) .

Для установки TensorFlow требуется версия pip 19 или выше. Введите следующую команду, чтобы обновить pip до последней версии:

pip install --upgrade pip

Теперь, когда виртуальная среда активирована, пришло время установить пакет TensorFlow.

pip install --upgrade tensorflow

Если у вас есть выделенный графический процессор NVIDIA и вы хотите использовать преимущества его вычислительной мощности, вместо tensorflow , установите пакет tensorflow-gpu , который включает поддержку графического процессора.

В виртуальной среде вы можете использовать команду pip вместо pip3 и python вместо python3 .

Это оно! Вы успешно установили TensorFlow и можете начать его использовать.

Чтобы проверить установку, выполните следующую команду, которая распечатает версию TensorFlow:

python -c 'import tensorflow as tf; print(tf.__version__)'

На момент написания этой статьи последней стабильной версией TensorFlow была 2.2.0:

2.2.0

Ваша версия TensorFlow может отличаться от версии, показанной здесь.

Если вы новичок в TensorFlow, посетите страницу « Начало работы с TensorFlow» и узнайте, как создать свое первое приложение машинного обучения. Вы также можете клонировать репозитории TensorFlow Models или TensorFlow- examples из Github, а также изучить и протестировать примеры TensorFlow.

Как только вы закончите свою работу, деактивируйте среду, набрав deactivate , и вы вернетесь в свою обычную оболочку.

deactivate

Выводы

Мы показали вам, как установить TensorFlow в виртуальной среде на Ubuntu 20.04.

Если вы столкнулись с проблемой или хотите оставить отзыв, оставьте комментарий ниже.

Оставьте комментарий