Символические тензоры заменяют мои обычные тензоры на Tensorflow-GPU - PullRequest
Новая
       52

Символические тензоры заменяют мои обычные тензоры на Tensorflow-GPU

0 голосов
/ 21 октября 2019

Недавно я установил тензор-gpu вместо tenorflow-cpu, и мой старый код больше не работает. При загрузке изображений вместо наличия тензоров определенной формы он дает мне «символические тензоры» с формой (?,?,?) И без значений в них, которые я, очевидно, не могу использовать для визуализации своих изображений.

Я пытался искать вещи, но я действительно не понимаю, почему мой код больше не работает, так как единственное, что я изменил, это тензор потока. 

input_image = tf.io.read_file("./dataset/smth/smth/0000.jpg")
input_image = tf.image.decode_jpeg(input_image)
input_image = tf.cast(input_image, tf.float32)
<tf.Tensor 'Cast_14:0' shape=(?, ?, ?) dtype=float32>

Что я получаю:

+ "Невозможно преобразовать символический тензор (truediv_1: 0) в массив с нулевыми значениями"

Что я хочу:

Также вотсписок моих устройств: 

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 14952830314257248667
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3225865420
locality {
  bus_id: 1
  links {
  }
}
incarnation: 6839329622363795927
physical_device_desc: "device: 0, name: GeForce GTX 1050 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1"
]
...