Tensorflow C++ api 开发例一

最新推荐文章于 2025-03-19 22:32:37 发布

fox_rabbit

最新推荐文章于 2025-03-19 22:32:37 发布

阅读量1.5k

点赞数

分类专栏： Tensorflow C++ API

本文链接：https://blog.youkuaiyun.com/abddb588/article/details/82500574

版权

本文档介绍了如何使用TensorFlow C++ API进行开发，包括编写源文件`test.cc`，创建Session并检查其创建是否成功。CMakeLists配置涉及到链接库的路径，如TensorFlow动态库、Eigen3、nsync和protobuf。在编译过程中，`bazel-genfiles`目录下的头文件对于API的正常使用至关重要，打包时需注意包含这一路径，确保所有依赖的头文件完整。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在获得libtensorflow_cc.so和libtensorflow_framework.so库后，开始进行C++api开发

编写源文件
源文件：test.cc

#include "/home/zhouxd/tools/tensorflow-master/tensorflow/core/public/session.h"
//tensorflow-master为tensorflow源码根目录，不喜欢include这么长的路径可以把这个路径加到cmakelist,再直接包含session.h
#include <iostream>
using namespace std;
using namespace tensorflow;
int main()
{
    //仅仅为了测试Session对象能否成功建立
    Session* session;
    Status status = NewSession(SessionOptions(), &session);
    if (!status.ok())