“Obtenez le nom du périphérique GPU TensorFlow” Réponses codées

Obtenez le nom du périphérique 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 vérifie mon 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

Réponses similaires à “Obtenez le nom du périphérique GPU TensorFlow”

Questions similaires à “Obtenez le nom du périphérique GPU TensorFlow”

Plus de réponses similaires à “Obtenez le nom du périphérique GPU TensorFlow” dans Python

Parcourir les réponses de code populaires par langue

Parcourir d'autres langages de code

Shell/Bash

C++

CSS

HTML

Java

JavaScript

Objective-C

PHP

Python

Sql

Swift

Ruby

TypeScript

Kotlin

Assembly

VBA

Scala

Rust

Dart

Elixir

Clojure

Haskell

Matlab

Erlang

Cobol

Fortran

Scheme

Perl

Groovy

Lua

Julia

Delphi

Abap

Lisp

Prolog

Pascal

ActionScript

Basic

Solidity

PowerShell

GDScript

Excel