How to list all tensorflow devices

This is how you do:

The basic installation of TensorFlow from the pip is typically done for the CPU version of Tensorflow. However you might need to install the CUDA support for Tensorflow. In this case, if you want to check if your GPU installation is successful.

from tensorflow.python.client import device_lib

def get_available_devices():
    local_device_protos = device_lib.list_local_devices()
    return [x.name for x in local_device_protos if x.device_type == 'GPU' or x.device_type=='CPU']
get_available_devices()