I am trying to run a tensorflow code in a Kaggle notebook. I want to use the gpu which is enabled. I tried the following code but the model fitting is using the cpu instead of t