缺少:tensorflow.compiler.xla.service导入hlo_pb2

时间:2019-05-05 14:43:19

标签: python tensorflow nvidia-jetson tensorrt

我正在尝试构建一个TensorRT文件以在我的jetson nano上运行。 我正在运行tensorflow 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。 在'hlo_xxxxxx'位置有tensorflow.compiler.xla.service个文件分配,但没有"hlo_pb2"

有什么想法吗?

0 个答案:

没有答案