如何构建和使用Google TensorFlow C++ api

2025-05-15 05:03:44
推荐回答(1个)
回答(1):

刚开始的话,先要从Github下载源代码,可以按这里的说明来操作从源码安装,需要注意的是:安装之前,应该要有Bazel和最近版本的GCC)。
Tensorflow C++ API(以及系统的后端)在tensorflow/core这个模块中,目前只支持2中形式:C++ Session interface(C++会话接口)和C API。可以使用这2中之一来执行TensorFlow图(使用Python API构建的序列化为GraphDef协议缓冲区)。还有一个用C++构建图形的实验功能,但是目前它并不像Python API那样像支持全部功能(full-featured),例如目前不支持auto-differentiation。可以看到builds a small graph in C++ here(在C++中构建简单的Graph)的示例程序。
C++ API的第二部分是用于添加新的OpKernel的API,它是包含用于CPU和GPU的数字内核的实现的类。在tensorflow/core/kernels以及tutorial for adding a new op in C++中有大量关于如何构建的示例