← Events and Videos
May 21, 2021 | 11:30AM ET

Using Dask and many GPUs to train a neural network with PyTorch

Jacqueline Nolis, Saturn Cloud

I took a neural network I had trained on a single CPU to generate pet names and tried retraining it with tons of connected GPUs using Dask, PyTorch, and the package dask-pytorch-ddp. I learned a lot about when is the right time to use multiple GPUs and what the pitfalls can be. In this talk I'll discuss what these lessons mean for training with GPUs and Dask.