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 !
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 !