Chapter 5's rnn-gluon notebook to use GPU.

zackchase / mxnet-the-straight-dope

An interactive book on deep learning. Much easy, so MXNet. Wow. [Straight Dope is growing up] ---> Much of this content has been incorporated into the new Dive into Deep Learning Book available at https://d2l.ai/.

https://d2l.ai/

Apache License 2.0

2.56k stars 724 forks source link

Chapter 5's rnn-gluon notebook to use GPU. #540

Closed vishaalkapoor closed 5 years ago

vishaalkapoor commented 5 years ago

The rnn gluon notebook takes much too long on the CPU. It takes ~ 5 minutes on a p3.8xl using one of the GPUs.