“Vérifiez le GPU dans TensorFlow” Réponses codées

Vérifiez si TensorFlow GPU est installé

import tensorflow as tf
print(tf.test.gpu_device_name())
CBT fan club

Test de GPU TensorFlow

# As it's written in Tensorflow documentatoin: 

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
Ugly Unicorn

Tensorflow Check GPU

tf.config.list_physical_devices('GPU')
Splendid Scarab

Vérifiez le GPU dans TensorFlow

# For tensorflow 2:
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

# For tensorflow 1:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Lazy Lizard

Test de GPU TensorFlow

tf.test.is_gpu_available(
  cuda_only=False, min_cuda_compute_capability=None
)
Witty Whale

Tensorflow ne peut pas voir GPU

$ pip uninstall tensorflow
$ pip install tensorflow-gpu
Relieved Reindeer

Réponses similaires à “Vérifiez le GPU dans TensorFlow”

Questions similaires à “Vérifiez le GPU dans TensorFlow”

Plus de réponses similaires à “Vérifiez le GPU dans TensorFlow” dans Python

Parcourir les réponses de code populaires par langue

Parcourir d'autres langages de code