我正在尝试构建一个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"
。
有什么想法吗?