Deep Learning Lab 4: Neural Network on GPU

Description

The goal of this lab is to see the improvement that we can get by training a neural network with a GPU instead of a CPU. For this, we rely on the Google colab environment that allow to run Jupyter notebooks on both CPU and GPU. Therefore we will see how to specify to PyTorch, the device to use for calculation. We will use the MNIST data set for this lab.

Here is the link to access the notebook on Google colab. It doesn’t work directly on github.