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

0 vote151 views1 answer
Python display map
Python
0 vote150 views1 answer
Calculate angle between 3 points python
Python
0 vote146 views1 answer
Django queryset to instance
Python
0 vote191 views1 answer
Pandas copy data from a column to another
Python
0 vote112 views1 answer
5+2 in python
Python
0 vote155 views1 answer
Df.min python
Python
0 vote136 views1 answer
Python ascii art
Python
0 vote145 views2 answers
How to play a mp3 file in python
Python
0 vote160 views1 answer
Swapping upper case and lower case string python
Python
2 votes140 views3 answers
Relative path python
Python