NVIDIA и Facebook ускоряют фреймворк глубокого обучения Caffe2

NVIDIA и Facebook представили результаты совместной работы в области искусственного интеллекта вместе с анонсом Caffe2 – нового фреймворка глубокого обучения с открытым кодом, созданного Facebook.

Каждый день в мире генерируется новая информация – тексты, картинки, видео и другие формы контента. Facebook разрабатывает новые системы искусственного интеллекта для работы с данной информацией, чтобы, несмотря на лавинообразный рост создаваемого контента, люди могли лучше понимать мир и более эффективно в нем взаимодействовать. Caffe2 позволяет разработчикам и исследователям создавать сценарии масштабного распределенного обучения и приложения машинного обучения для конечных устройств.

Предоставление сервисов на базе искусственного интеллекта на мобильных устройствах, несмотря на сложность процесса, должно происходить за доли секунды. Для мгновенной обработки запросов требуются графические процессоры, обеспечивающие молниеносную обработку данных, например, такие, как установлены в серверах Facebook Big Basin, а также оптимизированное программное обеспечение для задач глубокого обучения, которое способно реализовать весь потенциал аппаратной платформы.

NVIDIA и Facebook удалось ускорить работу ИИ в рамках фреймворка глубокого обучения Caffe2. Совместными усилиями компании оптимизировали работу Caffe2 с помощью платформы глубокого обучения на базе графических процессоров NVIDIA. Caffe2 использует новейшие SDK-библиотеки NVIDIA — cuDNN, cuBLAS и NCCL – обеспечивая быстрое обучение и инференс с ускорением на мульти-GPU конфигурациях. В результате, разработчики могут сосредоточиться на создании ИИ-приложений, зная, что Caffe2 обеспечит максимальную производительность на их системах на базе графических процессоров NVIDIA.

Caffe2 – это быстрый, масштабируемый и портативный фреймворк глубокого обучения. Он обеспечивает почти линейное масштабирование обучения глубоких сетей с ускорением пропускной способности в 57 раз на восьми объединенных в сеть серверах Facebook Big Basin с 64 ускорителями NVIDIA Tesla P100. Это означает, что разработчики могут обучать и итерировать ИИ-модели быстрее, чем когда бы то ни было.

В рамках сотрудничества компаний суперкомпьютер NVIDIA DGX-1 станет первой системой для ИИ с поддержкой Caffe2 в составе программного стека, оптимизированного для задач глубокого обучения. DGX-1 вместе с Caffe2 обеспечивают высокую производительность и быстрое обучение. Caffe2 для DGX-1 станет доступен для пользователей через NVIDIA DGX-1 Container Registry.

Обучение Caffe2 в Институте глубокого обучения NVIDIA

В рамках «Института глубокого обучения NVIDIA» (Deep Learning Institute - DLI) уже более 10 000 разработчиков по всему миру научились применять фреймворки для создания, обучения и развертывания нейронных сетей машинного обучения для разнообразных приложений и сервисов.

После конференции NVIDIA по GPU-технологиям (GTC), которая пройдет с 8 по 11 мая, курс по Caffe2 будет добавлен в расписание Института глубокого обучения (DLI) . Команда DLI будет помогать разработчикам применять Caffe2 в своей лаборатории.