Атрибут device класса Variable
Атрибут device класса Variable возвращает строку,
содержащую имя устройства, на котором размещена переменная.
Устройство определяет, где физически хранятся данные переменной:
на центральном процессоре (CPU) или на графическом процессоре (GPU).
Атрибут доступен только для чтения и не принимает параметров.
Синтаксис
v.device
Пример
Давайте создадим переменную и выведем имя устройства, на котором она размещена:
import tensorflow as tf
v = tf.Variable([1, 2, 3, 4, 5])
print(v.device)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте создадим переменную внутри области видимости
устройства CPU:0 и проверим значение атрибута:
import tensorflow as tf
with tf.device('/CPU:0'):
v = tf.Variable([1, 2, 3, 4, 5])
print(v.device)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте создадим переменную с явным указанием устройства
через параметр device:
import tensorflow as tf
v = tf.Variable([1, 2, 3, 4, 5], device='/CPU:0')
print(v.device)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
Пример
Давайте проверим, совпадает ли устройство переменной с устройством тензора, созданного из нее:
import tensorflow as tf
v = tf.Variable([1, 2, 3, 4, 5])
t = tf.constant([1, 2, 3, 4, 5])
print(v.device)
print(t.device)
Результат выполнения кода:
"/job:localhost/replica:0/task:0/device:CPU:0"
"/job:localhost/replica:0/task:0/device:CPU:0"
