How to use GPU on Jupyter notebook in Google Cloud AI Platform

Question

I\'m using a GPU-enabled Jupyter notebook on the Google Cloud AI Platform.

However, despite being GPU enabled, it is only using the CPU when I train my machine learning m