...
The last function call sets the GPU Tensorflow will use for each process. Try the script using distributed_tf.sh. job script using runTensorflow.sh provided in the previous page (adapt it for the use of this python script) You should see an output similar to the following one.
...