Did you solve this issue? My models are also training on CPU, not GPU. Tensorflow does not recognize the GPU from Jupyter or console.
I created an environment from lambda stack following the instructions in this forum. Pip list does show tensorflow-gpu in the virtual environment.