Я пытаюсь создать файл TensorRT для запуска на моем jetson nano.
Я бегу tenorflow 1.13.
Когда я запускаю этот код:
trt_graph = trt.create_inference_graph(
input_graph_def=['input_1'],
outputs=['Logits/Softmax'],
max_batch_size=1,
max_workspace_size_bytes=1 << 25,
precision_mode='FP16',
minimum_segment_size=50
)
Я получаю эту ошибку:
ImportError: cannot import name hlo_pb2
Он говорит, что ищет здесь:
from tensorflow.compiler.xla.service import hlo_pb2
Это существует?
Я обновил тензор потока.
В папке tensorflow.compiler.xla.service имеется множество файлов 'hlo_xxxxxx', но нет "hlo_pb2".
Есть идеи?