Run the ipynb file on google colab with GPU on. On running the second cell, an error will appear and an option showing to restart runtime. Click on that button and wait for 5 seconds. Then start running from the first cell again. This time there will be no error on the second cell. After this run each block in order.