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

29 votes333 views4 answers
How to remove all characters from a string in python
Python
0 vote371 views1 answer
Install python on windows subsystem for linux
Python
11 votes348 views3 answers
Python pandas change column order
Python
1 vote346 views1 answer
The four pillars of Op in Python
Python
6 votes354 views3 answers
How to remove all zeros from a list in python
Python
4 votes375 views4 answers
Filter django or
Python Django
8 votes349 views2 answers
Numpy.where
Python
10 votes343 views3 answers
Add picture to jupyter notebook
Python
6 votes363 views1 answer
How to get random int between two numbers python
Python
10 votes339 views1 answer
Pandas select columns where value is true
Python