I am trying to use my GPU with TensorFlow on a simple MNIST model. I have installed CUDA 10.1 and cuDNN 7.6.5. It seems to work(the model is faster than before, 2 seconds an