Get device name tensorflow

from tensorflow.python.client import device_lib

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

Similar codes

0 vote122 views1 answer
Where to import render in django
Python
0 vote119 views1 answer
Loi normale python numpy
Python
16 votes112 views1 answer
Super idol
Python
16 votes140 views5 answers
Primes in python
Python
17 votes110 views3 answers
Making log files in python
Python
0 vote166 views1 answer
Python trim zero off end of list
Python
0 vote106 views1 answer
Python draw polygon
Python
0 vote145 views1 answer
Assign elements of list to variables python
Python
8 votes170 views2 answers
Flask.pocoo.org quickstart
Python Flask
0 vote119 views1 answer
How to check if string is in byte formate pythin
Python