tocdepth

2

Быстрый старт

ML Space — платформа для ML-разработки полного цикла и совместной работы DS-команд над созданием и развертыванием моделей машинного обучения.

Платформа ML Space предоставляет пользователям решать следующие задачи:

  • Хранение и обработка артефактов машинного обучения.

  • Распределенное обучение на большом количество GPU.

  • Высокопроизводительные вычисления на базе CPU.

  • Развертывание (инференс) моделей машинного обучения.

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

В быстром старте описываются действия, которые будем выполнять через интерфейс платформы.

Перед началом работы

Убедитесь, что:

Работа с данными

На платформе представлено два типа хранилищ:

  • Объектное хранилище S3 — основной ресурс для хранения большого объема данных. Общее для платформы.

  • NFS — обеспечивает быстрый доступ к данным во время обучения моделей. В каждом регионе свое NFS-хранилище.

Чтобы использовать данные на платформе:

  1. Откройте вкладку Объектное хранилище Кнопка Объектное хранилище в левом меню и выберите бакет или создайте пользовательский.

  2. Загрузите данные в объектное хранилище S3, используя кнопку Создать или Data transfer service Кнопка сервиса Data transfer service.

  3. Чтобы использовать данные для обучения, переложите их в «горячее» хранилище NFS, используя кнопку Кнопка Отправить на NFS (находится в правом верхнем углу экрана) в объектном хранилище S3 или другие способы переноса.

Обучение моделей

На платформе доступно несколько способов обучения моделей.

  • Создание окружения (Jupyter Server) и интерактивная работа в нем. Код исполняется на выделенных заранее ресурсах с GPU/CPU.

  • Обучение через отправку задачи обучения в регион размещения вычислительных ресурсов. Для этого используется Jupyter Server Управление распределенными задачами.

В примере рассмотрим обучение с использованием Jupyter Server:

  1. Откройте Environments → Jupyter Servers → Создать Jupyter Server и создайте Jupyter Server с одним GPU. Подробнее в разделе Создать Jupyter Server.

  2. Подключитесь к Jupyter Server. В Jupyter Server для каждого региона используется отдельное NFS-хранилище.

  3. Запустите код.

Если необходимо запустить распределенное обучение (Pytorch DDP или Horovod), создайте Jupyter Server для управления распределенными задачами и отправьте задачу через Функция client_lib или воспользуйтесь примером с GitHub.

Установка дополнительных библиотек в Jupyter Server

Для установки дополнительных библиотек выполните команду в ячейке ноутбука:

!pip install <package_name> <version>

Где package_name — наименование библиотеки, которую предполагается установить, а version — версия данной библиотеки.

После установки библиотеки выполните следующую команду для проверки:

!pip list | grep <package_name>

Примечание

В Jupyter Server есть каталоги, в которых хранятся служебные и пользовательские файлы:

  • /home/user — каталог, уникальный для каждого Jupyter Server.

  • /home/jovyan — каталог, общий для всех Jupyter Server, созданных в рамках одного воркспейса.

Если устанавливать библиотеки с помощью команды pip install, то зависимости будут установлены в каталог /home/jovyan/.img-xxxxx.

При постановке Jupyter Server на паузу этот каталог остается, при остановке (удалении) каталог удаляется. При удалении Jupyter Server все библиотеки, которые установлены с помощью команды pip install, удаляются вместе с каталогом /home/user.

Для использования требуемого набора библиотек можно создать и использовать кастомный Docker-образ. Подробнее см. Способ 1. Обучение из Jupyter Server с GPU.

Больше примеров работы с платформой

Примеры по обучению и использованию моделей размещены на GitHub. Среди них:

  • pytorch-example — задача распределенного обучения Pytorch-модели с двумя типами запуска: стандартный horovod и дополнительный pytorch, он же Pytorch.Distributed.

  • hugging-face-llm-example — работа с языковой моделью методами LoRA и PEFT, а также распределенное обучение с PyTorch Distributed Data Parallel (DDP).

  • lightning-example — использование PyTorch и PyTorch Lightning для задачи классификации изображений.

  • pytorch-elastic-example — обучение модели на PyTorch и Elastic Learning с сохранением контрольных точек обучения (чекпоинтов).

Запустили Evolution free tier
для Dev & Test
Получить