get device name tensorflow

Solutions on MaxInterview for get device name tensorflow by the best coders in the world

showing results for - "get device name tensorflow"
Cameron
10 Sep 2018
1from tensorflow.python.client import device_lib
2
3def get_available_gpus():
4    local_device_protos = device_lib.list_local_devices()
5    return [x.name for x in local_device_protos if x.device_type == 'GPU']