TensorFlow – это мощная библиотека с открытым исходным кодом, разработанная Google для задач машинного обучения, которая широко используется исследователями, разработчиками и специалистами по изучению данных для построения и обучения моделей машинного обучения.
Если вы используете машину Ubuntu и хотите установить TensorFlow, это руководство проведет вас через процесс установки и использования TensorFlow в вашей системе.
Предварительные условия
Перед установкой TensorFlow убедитесь, что ваша машина Ubuntu соответствует следующим требованиям:
- Python: TensorFlow поддерживает Python 3.7-3.10.
- Pip: Pip – это менеджер пакетов для Python.
- Оборудование: Хотя TensorFlow может работать на CPU, наличие GPU может значительно ускорить обучение больших моделей.
Шаг 1: Установите Python и Pip в Ubuntu
Перед установкой любого нового программного обеспечения стоит обновить список пакетов и обновить установленные пакеты.
sudo apt update sudo apt upgrade
Далее установите Python и pip, менеджер пакетов для Python, используемый для установки и управления пакетами Python.
sudo apt install python3 sudo apt install python3-pip
После установки вы можете проверить установку Python и pip перед установкой таких пакетов, как TensorFlow.
python3 --version pip3 --version
Проверьте версию Python и PIP
Шаг 2: Создание виртуальной среды в Ubuntu
Создание виртуальной среды необязательно, но рекомендуется, так как это поможет сохранить вашу установку TensorFlow изолированной от других проектов Python.
Создайте новый каталог для проекта TensorFlow и перейдите в него:
mkdir tensorflow_project cd tensorflow_project
Создайте виртуальную среду и активируйте ее:
python3 -m venv tensorflow_env source tensorflow_env/bin/activate
Теперь в приглашении терминала должно отображаться имя виртуальной среды, что указывает на то, что она активна.
Создайте каталог проекта Tensorflow
Шаг 3: Установите TensorFlow в Ubuntu
Теперь, когда ваше окружение настроено, вы можете установить последнюю стабильную версию TensorFlow вместе с его зависимостями, используя pip, менеджер пакетов Python.
pip install --upgrade pip pip install tensorflow
Установите TensorFlow в Ubuntu
Шаг 4: Проверка TensorFlow в Ubuntu
После завершения установки стоит убедиться, что TensorFlow установлен правильно, зайдя в оболочку Python.
python3
Импортируйте TensorFlow и проверьте его версию:
import tensorflow as tf print(tf.__version__)
Если TensorFlow установлен правильно, то номер версии будет выведен без ошибок.
2.18.0
Вы также можете запустить простой тест, чтобы убедиться, что TensorFlow работает.
hello = tf.constant('Hello, TensorFlow!')
print(hello)
В результате должно получиться что-то вроде
tf.Tensor(b'Hello, TensorFlow!', shape=(), dtype=string)
Шаг 5: Установите TensorFlow с поддержкой GPU (необязательно)
Если у вас есть совместимый NVIDIA GPU и вы хотите использовать его для более быстрых вычислений, вы можете установить TensorFlow с поддержкой GPU.
lspci | grep -i nvidia sudo apt install nvidia-driver-535 -y sudo reboot
Далее установите TensorFlow с поддержкой GPU.
pip install tensorflow-gpu
Убедитесь в правильности установки, проверив, обнаруживает ли TensorFlow ваш GPU:
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
Заключение
Вы успешно установили TensorFlow на Ubuntu 24.04. Независимо от того, работаете ли вы над простым проектом машинного обучения или сложной моделью глубокого обучения, TensorFlow предоставляет инструменты, необходимые для начала работы.
Не забывайте об организации среды с помощью виртуальных окружений и обратите внимание на поддержку GPU, если вам нужны более быстрые вычисления.
Комментарии (0)