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_tensorflow
cd 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.
Если вы столкнулись с проблемой или хотите оставить отзыв, оставьте комментарий ниже.