IBM и Nvidia смогли ускорить глубокое обучение нейронных сетей

Компании IBM и Nvidia рассказали о сотрудничестве, направленном на ускорение глубокого обучения нейронных сетей. Предложенное партнерами решение включает программное обеспечение IBM PowerAI, работающее на специализированном сервере IBM Power System S822LC на процессорах IBM Power8. Особенностью сервера является наличие интерфейса Nvidia NVLink. Как утверждается, это аппаратное средство прямого взаимодействия между CPU и GPU позволило более чем удвоить производительность по сравнению с сопоставимыми серверами с четырьмя GPU в тесте AlexNet для нейронной сети Caffe. Сеть Caffe, разработанная Berkeley Vision and Learning Center (BVLC), является одной из пяти, включенных в IBM PowerAI. Кроме того, IBM PowerAI включает библиотеки Nvidia GPUDL: cuDNN, cuBLAS и NCCL.

Секрет ускорения — в ПО IBM PowerAI Software Toolkit и интерфейсе Nvidia NVLink

Пакет IBM PowerAI уже доступен. ОН предложен бесплатно всем пользователями IBM Power S822LC.

Источник: IBM

Теги: IBM , Nvidia , NVLink

Комментировать

©  iXBT