Метод 1: Установка TensorFlow с помощью PiP

1. Установите обновления системы

Сначала запустим команду обновления системы на нашей системе Ubuntu 24.04, чтобы убедиться в стабильности и совместимости пакетов. Это также обновит список пакетов в репозитории.

sudo apt update && sudo apt upgrade -y

2. Установите Python, PIP и VENV

Для установки TensorFlow нам требуются Python и PIP, поэтому используйте данную команду для их установки, если они еще не были установлены. Лучше всего использовать виртуальное окружение, чтобы управлять зависимостями и избегать конфликтов.

  • Для Python3:
sudo apt install python3 python3-dev
  • Чтобы проверить версию:
python3 --version
  • Кроме Python, нам также нужен менеджер пакетов, ‘PIP.’ Чтобы установить его, используйте:
sudo apt install python3-pip
  • Установить Python venv чтобы получить виртуальное окружение
sudo apt install python3-venv

3: Создание виртуальной среды

Для установки TensorFlow на Ubuntu 24.04 и его зависимостей мы создаем виртуальное окружение, чтобы изолировать его зависимости от других проектов. Это хорошая практика, чтобы избежать конфликтов.

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

python3 -m venv tensorflow_env

Активируйте виртуальную среду:

source tensorflow_env/bin/activate

Теперь в приглашении на вашем терминале будет имя созданной среды.

4: Установка TensorFlow на Ubuntu 24.04 с помощью PIP

Активировав виртуальную среду, установите TensorFlow, используя pip. Сначала мы обновим PIP, а затем выполним команду для загрузки и настройки «TensorFlow

pip install --upgrade pip
pip install tensorflow

5. Чтобы проверить версию TensorFlow

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

pip show tensorflow
проверка версии TensorFlow

Метод 2: Установка TensorFlow с помощью Conda

Те, кто использует дистрибутив Python «Anaconda» можно установить Tensorflow с помощью Conda, менеджера пакетов Anaconda.

Шаг 1: Установка Anaconda на Ubuntu 24.04

Вы можете пропустить этот шаг, если у вас уже установлена Anaconda на вашей системе Ubuntu. Другие пользователи, напротив, могут ознакомиться с нашим руководством по загрузке инсталляции Anaconda с официального сайта для установки на Ubuntu 24.04 или любую другую предыдущую версию.

Шаг 2: Создание среды Conda (необязательно)

Сначала создайте окружение conda для вашего проекта TensorFlow. Вы можете даже выбрать пользовательскую версию Python для вашего проекта. Например, здесь мы используем v3.12. Также здесь мы используем «mytensorflow_env» в качестве имени нашего окружения. Вы можете задать ему любое значение.

conda create -n mytensorflow_env python=3.12

Активируйте среду conda:

После настройки окружения активируйте его, используя приведенный синтаксис.

conda activate mytensorflow_env

Шаг 3: Установка TensorFlow с помощью Conda

Теперь мы будем использовать Conda для установки Tensorflow в наше активированное окружение; вот команда, которую нам нужно выполнить:

conda install -c conda-forge tensorflow

В приведенной выше команде мы используем «Conda-Forge» — канал, поддерживаемый сообществом, который предлагает коллекцию различных последних версий пакетов для пакетного менеджера Conda. Это один из самых популярных каналов для получения актуальных и хорошо поддерживаемых пакетов в экосистеме Conda.

Однако если вы хотите использовать канал по умолчанию, поддерживаемый Anaconda, Inc, ниже будет использована та же команда без указания имени канала. Однако некоторые пакеты не всегда могут иметь последние версии так же быстро, как conda-forge.

conda install tensorflow

Тогда как для некоторые конкретные версии TensorFlow, мы можем указать версию в приведенной выше команде, например:

conda install tensorflow=2.14.0

Шаг 4: Проверьте доступность Tensorflow:

После завершения установки мы можем использовать команду для проверки наличия Tensorflow в нашей системе.

conda list | grep tensorflow

Дополнительно: Поддержка GPU

До сих пор мы видели, как установить Tensorflow без поддержки GPU. Однако если у вас есть графический процессор NVIDIA, вам следует включить его поддержку для повышения производительности. Прежде чем воспользоваться преимуществами ускорения GPU, в Ubuntu необходимо установить следующее дополнительное программное обеспечение.

  • Драйвер NVIDIA: Установите соответствующий драйвер NVIDIA для вашего GPU.
  • CUDA Toolkit: Установите набор инструментов CUDA с сайта сайт NVIDIA.
  • cuDNN: Скачайте и установите cuDNN с сайта сайта NVIDIA.

После установки вышеуказанных компонентов вы можете установить GPU-версию TensorFlow:

Использование Python PIP

pip install tensorflow-gpu

Или с помощью conda:

conda install -c conda-forge tensorflow-gpu

PIP и Conda — популярные способы установки фреймворка машинного обучения TensorFlow в системе Ubuntu Linux. После завершения процесса установки вы можете строить и развертывать модели машинного обучения, используя любой из двух методов, рассмотренных в этой статье.