vDNN: Virtualized Deep Neural Networks for Scalable, Memory-Efficient Neural Network Design

Preprint English OPEN
Rhu, Minsoo; Gimelshein, Natalia; Clemons, Jason; Zulfiqar, Arslan; Keckler, Stephen W.;
  • Subject: Computer Science - Distributed, Parallel, and Cluster Computing | Computer Science - Neural and Evolutionary Computing | Computer Science - Learning

The most widely used machine learning frameworks require users to carefully tune their memory usage so that the deep neural network (DNN) fits into the DRAM capacity of a GPU. This restriction hampers a researcher's flexibility to study different machine learning algori... View more
Share - Bookmark