Hi @Julia_Wasala Welcome to the TF Forum 



 Julia_Wasala:

I’m using tensorflow together with autokeras on GPU, it took some time to find matching versions that would recognize the GPU.


Can you please share your versions of CUDA, cuDNN, GPU driver, etc (دعم GPU  |  TensorFlow)



 Julia_Wasala:

Now I installed tensorflow-datasets, autokeras doesn’t work anymore.


Will try to chase the right team to look into this. Which version of Autokeras, TFDS and TensorFlow do you have right now?
As far as I understand the min version of TF for TFDS is 2.1: datasets/tf_compat.py at 93c1d96dee42b451038e9303166e4c7e83a26b82 · tensorflow/datasets · GitHub