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 vote216 views3 answers
Draw circle opencv
Python
0 vote207 views1 answer
Failed to build wxPython
Python
0 vote213 views2 answers
Python combine two lists into matrix
Python
-1 vote191 views1 answer
Add gaussian noise python
Python
0 vote209 views1 answer
Pyspark check all columns for null values
Python
1 vote208 views4 answers
Drop duplicate index pandas
Python
0 vote224 views1 answer
Turn false true column into 0 1 pandas
Python
3 votes214 views1 answer
Multiple linear regression in python
Python
0 vote219 views1 answer
Check permissions django template
Python
2 votes220 views1 answer
Add a list in python
Python