running your first notebook taking > 5 mins in google colab
Make sure you have GPU turned on in colab