tf.config.experimental.list_physical_devices

TensorFlow 1 version View source on GitHub

Return a list of physical devices visible to the runtime.

tf.config.experimental.list_physical_devices(
    device_type=None
)

Used in the notebooks

Used in the guide Used in the tutorials

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