#tensorflow
#tensorflow
Вопрос:
Я прочитал много вопросов и «руководств» о том, как понять, работает ли Tensorflow на GPU, но я все еще в замешательстве.
Я сделал снимок экрана моего сеанса, и я хотел бы понять, что происходит, и работает ли Tensorflow на GPU или CPU.
Ответ №1:
Роберто, попробуй это
import tensorflow as tf
if tf.test.gpu_device_name():
print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
print("Please install GPU version of TF")
вывод должен быть
Default GPU Device: /device:GPU:0
GPU 0 — это ваш GTX 860m
Комментарии:
1. Спасибо, я получаю
Please install GPU version of TFнекоторую другую информацию о журнале. Но я уже установил графический процессор tensorflow, не хотите ли взглянуть на скриншот?2. пожалуйста, откройте свою среду Tensorflow в консоли и введите
python pip list, затем сделайте снимок экрана всех пакетов.3. Казалось, что перезапуск решил проблему, извините, что потратил ваше время, но большое спасибо за помощь!
4. То же самое для вас! Приветствия!
Ответ №2:
попробуйте
tf.debugging.set_log_device_placement(True)
затем запустите что-нибудь из tensorflow, например
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
затем консоль напечатает устройство, на котором работает tf
