“Uzyskaj nazwę urządzenia GPU TENSORFLOW” Kod odpowiedzi

Uzyskaj nazwę urządzenia 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 Sprawdź mój 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

Odpowiedzi podobne do “Uzyskaj nazwę urządzenia GPU TENSORFLOW”

Pytania podobne do “Uzyskaj nazwę urządzenia GPU TENSORFLOW”

Więcej pokrewnych odpowiedzi na “Uzyskaj nazwę urządzenia GPU TENSORFLOW” w Python

Przeglądaj popularne odpowiedzi na kod według języka

Przeglądaj inne języki kodu

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