“Periksa GPU di TensorFlow” Kode Jawaban

Periksa apakah GPU TensorFlow diinstal

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

Tes TensorFlow GPU

# 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 Periksa GPU

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

Periksa GPU di 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

Tes TensorFlow GPU

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

Tensorflow tidak dapat melihat GPU

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

Jawaban yang mirip dengan “Periksa GPU di TensorFlow”

Pertanyaan yang mirip dengan “Periksa GPU di TensorFlow”

Lebih banyak jawaban terkait untuk “Periksa GPU di TensorFlow” di Python

Jelajahi jawaban kode populer menurut bahasa

Jelajahi bahasa kode lainnya