Get gpu device name tensorflow

Python check my gpu

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']

Source: stackoverflow.com

Get gpu device name tensorflow

import tensorflow as tf

tf.test.gpu_device_name()

# Bash
# $ CUDNN_H_PATH=$(whereis cudnn.h)
#$ cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR

Similar codes

2 votes146 views2 answers
'flask' is not recognized as an internal or external command, operable program or batch file.
Python
2 votes145 views1 answer
How do i print a list line by line in python
Python
1 vote165 views1 answer
Markers seaborn
Python
0 vote163 views1 answer
Python read file between two strings
Python
14 votes141 views1 answer
Python function to print random number
Python
39 votes178 views2 answers
String to list of characters python
Python
1 vote161 views1 answer
Heroku change python version
Python
10 votes231 views6 answers
How to show a progress bar in python
Python
0 vote151 views1 answer
Punto1
Python
1 vote148 views1 answer
How to find the neighbors of an element in matrix python
Python