“Obtenga el nombre del dispositivo GPU TensorFlow” Código de respuesta

Obtenga el nombre del dispositivo GPU TensorFlow

import tensorflow as tf

tf.test.gpu_device_name()

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

Python revisa mi 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']
Jittery Jay

Respuestas similares a “Obtenga el nombre del dispositivo GPU TensorFlow”

Preguntas similares a “Obtenga el nombre del dispositivo GPU TensorFlow”

Más respuestas relacionadas con “Obtenga el nombre del dispositivo GPU TensorFlow” en Python

Explore las respuestas de código populares por idioma

Explorar otros lenguajes de código