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

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

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

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

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

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