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

1 vote309 views1 answer
Nlargest heapq
Python
4 votes306 views2 answers
Python read entire file
Python
1 vote321 views1 answer
Python cv2.Canny()
Python
0 vote321 views1 answer
Python get file name without dir
Python
0 vote319 views1 answer
Python pool.map() function
Python
0 vote299 views1 answer
Change variable type python
Python
0 vote314 views1 answer
Pyevtk documentation writearraystovtk
Python
1 vote330 views2 answers
Program to print duplicates from a list of integers in python
Python
1 vote298 views1 answer
Jupyter no output cell
Python
0 vote365 views1 answer
Django validate_comma_separated_integer_list
Python Django