NVIDIA DGX-1: первый в мире суперкомпьютер для глубокого обучения

Компания NVIDIA плотно взялась за разработки в сфере искусственного интеллекта и ранее уже была представлена платформа для автономных автомобилей Drive PX 2, которую уже начали использовать многие компании в своих разработках. NVIDIA пошла дальше и представила первый в мире суперкомпьютер для глубокого обучения NVIDIA DGX-1. По утверждению компании, DGX-1 сравним по скорости обучения глубоких сетей с 250 серверами x86 на базе Intel Xeon и способен обеспечить искусственный интеллект всеми необходимыми вычислительными ресурсами. 

Суперкомпьютер построен на базе восьми модулей с GPU Pascal Tesla P100. Общая производительность системы DGX-1 на вычислениях с половинной (FP16) точностью будет достигать 170 терафлопс. Каждый из 8 GPU Pascal Tesla P100 имеет по 16 ГБ памяти HBM 2. В решении впервые для связи графических процессоров использован новый интерфейс компании — NVLink, который обеспечивает ультрабыстрый обмен данными между CPU и GPU и между разными GPU. Обещают от 5 до 12 раз прирост скорости по сравнению с традиционным внутренним соединением PCIe Gen3. Для кэширования данных используется SSD на 7 ТБ. Сетевой интерфейс — два порта 10GbE и четыре InfiniBand 100Gb. Пиковое потребление — 3.2 КВт.

Система NVIDIA DGX-1 включает полный набор программ, оптимизированных для глубокого обучения, который позволяет ученым и исследователям в области данных легко и быстро обучать глубокие нейронные сети. В набор ПО DGX-1 входит NVIDIA Deep Learning GPU Training System (DIGITS), интерактивная система для создания глубоких нейронных сетей (DNN), GPU-ускоряемая библиотека примитивов для создания DNN - NVIDIA CUDA Deep Neural Network (cuDNN) версии 5. Помимо этого, система включает оптимизированные версии нескольких широко используемых фреймворков глубокого обучения— Caffe, Theano и Torch. В США система NVIDIA DGX-1 будет продаваться с июня, цена — около $129 тыс.