...
There are two modules for tensorflow on the cluster for python and both are gpu aware. Use the gpu partition to take advantage of the speedup the the gpus provide.
tensorflow/11-python2.7
tensorflow/11-python3.5
You must use python3 instead of python if using the tensorflow/11-python3.5 module.