Missed TensorFlow World? Check out the recap. Learn more

tf.config.experimental.list_physical_devices

TensorFlow 2.0 version View source on GitHub

Return a list of physical devices visible to the runtime.

Aliases:

  • tf.compat.v1.config.experimental.list_physical_devices
  • tf.compat.v2.config.experimental.list_physical_devices
tf.config.experimental.list_physical_devices(device_type=None)

Physical devices are hardware devices locally present on the current machine. By default all discovered CPU and GPU devices are considered visible. The list_physical_devices allows querying the hardware prior to runtime initialization.

The following example ensures the machine can see at least 1 GPU.

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "No GPUs found."

Args:

  • device_type: (optional) Device type to filter by such as "CPU" or "GPU"

Returns:

List of PhysicalDevice objects