I am trying to train a Deep net using OpenAI code that runs without error on a server with Tensorflow 2 CPU. However, when I try to run the very same code on a GPU machine with
