arXiv · 1710.11351
ChainerMN: Scalable Distributed Deep Learning Framework
Abstract
One of the keys for deep learning to have made a breakthrough in various fields was to utilize high computing powers centering around GPUs. Enabling the use of further computing abilities by distributed processing is essential not only to make the deep learning bigger and faster but also to tackle unsolved challenges. We present the design, implementation, and evaluation of ChainerMN, the distributed deep learning framework we have developed. We demonstrate that ChainerMN can scale the learning process of the ResNet-50 model to the ImageNet dataset up to 128 GPUs with the parallel efficiency of 90%.
Explore related subjects
Keep this discovery
Takuya Akiba, Keisuke Fukuda, Shuji Suzuki. 2017-10-31. ChainerMN: Scalable Distributed Deep Learning Framework. https://arxiv.org/abs/1710.11351
Cite the original work for its findings. Save a collection to share your selection of sources.