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
)

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."