我正在尝试构建一个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。
在tensorflow.compiler.xla.service
位置有'hlo_xxxxxx'
个文件的分配,但没有"hlo_pb2"
有什么想法吗
目前没有回答
相关问题 更多 >
编程相关推荐