在 C++ 中创建图形时 Tensorflow 分段错误?

问题描述

我正在尝试使用以下代码在 tensorflow C++ 中创建图形:

 tensorflow::Sessionoptions options;
 TF_CHECK_OK(tensorflow::NewSession(options,&session));

 std::string graph = "./my_test_model-1000.Meta";
 tensorflow::MetaGraphDef graph_def;
 TF_CHECK_OK(ReadBinaryProto(tensorflow::Env::Default(),graph,&graph_def));
 TF_CHECK_OK(session->Create(graph_def.graph_def()));

获取最后一行的segmentation fault错误

注意事项:

  1. 模型保存在 tensorflow2.4 中,版本 2 功能禁用
  2. 在 C++ 中使用 2.4 版加载图形。

我尝试加载不同的图表,但仍然无法正常工作。

我也使用了 GDB。然后我才知道段错误发生在哪里。

在版本 2 中加载版本 1 图形时是否需要考虑任何兼容性问题? 这是错误

 I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
15049 4
Segmentation fault

有人可以帮我解决这个问题吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)