W3cubDocs

/TensorFlow Python

tf.contrib.eager.num_gpus

tf.contrib.eager.num_gpus()

Defined in tensorflow/python/eager/context.py.

Get the number of available GPU devices.

Returns:

The number of available GPU devices.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/eager/num_gpus