Метод 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
Метод 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. После завершения процесса установки вы можете строить и развертывать модели машинного обучения, используя любой из двух методов, рассмотренных в этой статье.