“Holen Sie sich den GPU -Gerätenamen TensorFlow” Code-Antworten

Holen Sie sich den GPU -Gerätenamen 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 überprüfe meine 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

Ähnliche Antworten wie “Holen Sie sich den GPU -Gerätenamen TensorFlow”

Fragen ähnlich wie “Holen Sie sich den GPU -Gerätenamen TensorFlow”

Weitere verwandte Antworten zu “Holen Sie sich den GPU -Gerätenamen TensorFlow” auf Python

Durchsuchen Sie beliebte Code-Antworten nach Sprache

Durchsuchen Sie andere Codesprachen

Shell/Bash
C#
C++
C
CSS
HTML
Java
JavaScript
Objective-C
PHP
Python
Sql
Swift
Ruby
TypeScript
Go
Kotlin
Assembly
R
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