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 votes108 views1 answer
Head or tail python
Python
0 vote106 views1 answer
Split at the second occurrence of the element python
Python
4 votes113 views1 answer
Python zip file open as text
Python
5 votes118 views1 answer
How to get a dataframe column as a list
Python
0 vote128 views1 answer
Plotly reverse y axis
Python
0 vote113 views1 answer
Find word position in string python
Python
2 votes113 views2 answers
Python curses for windows
Python
5 votes105 views1 answer
Show full pd dataframe
Python
0 vote105 views1 answer
Using csv module how to read perticular lines in csv
Python
4 votes125 views2 answers
Pandas most frequent value
Python