How to work with GPU ?

[lablebi96]

Hello, I am working with tensorflow-gpu from Docker hub. So my tensorflow version is tensorflow-gpu:latest ! Meanwhile, when I launch the GAN training it starts automatically on the CPU. Are there some args to put to the model so that it turns on the GPU ? Is there any update about this since the issue of uninstalling tensorflow and installing tensorflow-gpu ? Thanks in advance for your help !