“Dapatkan nama perangkat GPU TensorFlow” Kode Jawaban

Dapatkan nama perangkat 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 Periksa GPU saya

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

Jawaban yang mirip dengan “Dapatkan nama perangkat GPU TensorFlow”

Pertanyaan yang mirip dengan “Dapatkan nama perangkat GPU TensorFlow”

Lebih banyak jawaban terkait untuk “Dapatkan nama perangkat GPU TensorFlow” di Python

Jelajahi jawaban kode populer menurut bahasa

Jelajahi bahasa kode lainnya