* Increase GPU counting robustness

Added native Tensorflow method to check number of GPUs.

* docs

* missing import

* fix import