You may find our tutorials in our blog helpful in this regard, especially since we have a short series specifically for Tensorflow.
Also, here is a Python command to use Tensorflow to ensure Tensorflow can see your GPUs:
python -c "import tensorflow as tf; print('Num GPUs Available: ', len(tf.config.experimental.list_physical_devices('GPU')))"
Otherwise you’ll need to edit your code/model to ensure the GPU(s) are being utilized.
If this doesn’t answer your question, let us know here or you can also open a support ticket here:
Linux Support Engineer