TensorFlow - это платформа с открытым исходным кодом для машинного обучения, созданная компанией Google. Она может работать на CPU или GPU на различных устройствах, и ее используют многие организации, включая Twitter, PayPal, Intel, Lenovo и Airbus.
TensorFlow можно установить в рамках всей системы, в виртуальной среде Python, как контейнер Docker или с помощью Anaconda.
TensorFlow поддерживает как Python 2, так и 3. Мы будем использовать Python 3 и установим TensorFlow в виртуальной среде. Виртуальная среда позволяет вам иметь несколько различных изолированных сред Python на одном компьютере и устанавливать определенную версию модуля для каждого проекта, не беспокоясь о том, что это повлияет на другие проекты.
Установка TensorFlow в CentOS
В отличие от других дистрибутивов Linux, Python не установлен по умолчанию на CentOS 8. Чтобы установить Python 3 на CentOS 8, выполните следующую команду в терминале от имени пользователя root или sudo:
1 | sudo dnf install python3 |
Команда выше установит Python 3.6 и pip . Чтобы запустить Python 3, вам нужно явно ввести python3, а чтобы запустить pip, введите pip3.
Начиная с Python 3.6, рекомендуемым способом создания виртуальной среды является использование модуля venv.
Перейдите в каталог, где вы хотите хранить ваш проект TensorFlow. Это может быть ваш домашний каталог или любой другой каталог, где у пользователя есть права на чтение и запись.
Создайте новый каталог для проекта TensorFlow и перейдите в него:
1 2 | mkdir tensorflow_project cd tensorflow_project |
Внутри каталога выполните следующую команду для создания виртуальной среды:
1 | python3 -m venv venv |
Приведенная выше команда создает каталог с именем venv, содержащий копию двоичного файла Python, pip стандартной библиотеки Python и другие вспомогательные файлы. Вы можете использовать любое имя для виртуальной среды.
Чтобы начать использовать виртуальную среду, активируйте ее, набрав:
1 | source venv/bin/activate |
После активации каталог bin виртуальной среды будет добавлен в начало переменной $PATH. Также изменится приглашение вашей оболочки, и в нем появится имя виртуальной среды, которую вы используете в данный момент. В данном случае это venv.
Для установки TensorFlow требуется pip версии 19 или выше. Выполните следующую команду, чтобы обновить pip до последней версии:
1 | pip install --upgrade pip |
Теперь, когда виртуальная среда создана и активирована, установите библиотеку TensorFlow с помощью следующей команды:
1 | pip install --upgrade tensorflow |
Если у вас есть выделенный графический процессор NVIDIA и вы хотите воспользоваться его вычислительной мощностью, вместо tensorflow установите пакет tensorflow-gpu, который включает поддержку GPU.
В виртуальной среде вы можете использовать команды pip вместо pip3 и python вместо python3.
Чтобы проверить установку, выполните следующую команду, которая выведет версию TensorFlow:
1 | python -c 'import tensorflow as tf; print(tf.__version__)' |
После завершения работы деактивируйте среду, набрав deactivate, и вы вернетесь к своей обычной оболочке.
1 | deactivate |