“Получите название устройства GPU TensorFlow” Ответ

Получите название устройства 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 проверьте мой графин

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

Ответы похожие на “Получите название устройства GPU TensorFlow”

Вопросы похожие на “Получите название устройства GPU TensorFlow”

Больше похожих ответов на “Получите название устройства GPU TensorFlow” по Python

Смотреть популярные ответы по языку

Смотреть другие языки программирования