Шаг 1. Подготовка вашей системы.
Для начала откройте окно терминала и обновите свою систему, чтобы убедиться, что у вас установлены последние пакеты и исправления безопасности. Выполните следующую команду:
sudo apt update sudo apt upgrade
Далее установите необходимые зависимости для сборки TensorFlow из исходного кода. Выполните следующие команды:
sudo apt install build-essential sudo apt install python3-dev python3-venv
Эти команды установят основные инструменты сборки и библиотеки разработки Python, необходимые для плавного процесса установки.
Шаг 2. Настройка среды Python.
Когда система подготовлена, пришло время настроить виртуальную среду Python. Виртуальные среды позволяют создавать изолированные экземпляры Python, предотвращая конфликты между различными проектами и их зависимостями. Чтобы установить Python и PIP, запустите:
sudo apt install python3 python3-pip
Теперь создайте новую виртуальную среду для TensorFlow, используя следующую команду:
python3 -m venv tensorflow_env
Это создаст новый каталог с именем tensorflow_env в вашем текущем местоположении. Чтобы активировать виртуальную среду, используйте:
source tensorflow_env/bin/activate
Теперь должно появиться приглашение терминала (tensorflow_env), указывая, что вы работаете в виртуальной среде.
Шаг 3. Установка TensorFlow в Ubuntu 24.04.
После активации виртуальной среды вы можете приступить к установке TensorFlow. Если вы используете систему только с процессором, запустите:
pip install tensorflow
Для систем с поддержкой графического процессора вам необходимо убедиться, что CUDA и cuDNN установлены правильно. Чтобы проверить версию CUDA, используйте:
nvcc --version
Чтобы убедиться, что ваш графический процессор распознан системой, запустите:
nvidia-smi
Если все настроено правильно, вы можете установить TensorFlow с поддержкой графического процессора, используя:
pip install tensorflow[and-cuda]
Эта команда автоматически загрузит и установит соответствующие версии CUDA и cuDNN, совместимые с вашей установкой TensorFlow.
Чтобы убедиться, что TensorFlow установлен правильно, запустите простой тестовый скрипт. Для систем только с процессором выполните:
python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
Если у вас система с поддержкой графического процессора, запустите:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
Если установка прошла успешно, вы должны увидеть результат вычислений или список доступных графических процессоров соответственно.
