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

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

Компании 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 уже доступен. ОН предложен бесплатно всем пользователями IBM Power S822LC.

Источник: IBM