tkschmidt / hello_tf_c_api

Neural Network TensorFlow C API

Example TensorFlow C API

Branch	Linux/OSX	Windows	License	Codacy
master

Example how to run TensorFlow lib C API on Windows, Linux and macOS(Darwin).

Example

• Hello TF
• Load graph
• Create Tensor
• Allocate Tensor
• Run session
• Interface
• Tensor Info
• Graph Info

Build example

Windows

git clone --depth 1 https://github.com/Neargye/hello_tf_c_api
cd hello_tf_c_api
mkdir build
cd build
cmake -G "Visual Studio 15 2017" -A x64 ..
cmake --build . --config Debug

Linux and macOS(Darwin)

git clone --depth 1 https://github.com/Neargye/hello_tf_c_api
cd hello_tf_c_api
mkdir build
cd build
cmake -G "Unix Makefiles" ..
cmake --build .

Remarks

• After the build, you can find the TensorFlow lib in the folder hello_tf_c_api/tensorflow/lib, and header in hello_tf_c_api/tensorflow/include.
• The tensorflow in the repository is compiled in x64 mode. Make sure that project target 64-bit platforms.
• Make sure that the tensorflow lib is in Output Directory or either in the directory contained by the %PATH% environment variable.

Get tensorflow lib

For x64 CPU, you can download the tensorflow.so, tensorflow.dll and tensorflow.lib from https://github.com/Neargye/tensorflow/releases.

Or build lib which version you need from the sources, with CPU or GPU support.

Link tensorflow lib

CMakeLists.txt

link_directories(yourpath/to/tensorflow) # path to tensorflow lib
... # other
target_link_libraries(<target> <PRIVATE|PUBLIC|INTERFACE> tensorflow)

Visual Studio

"Project"->"Properties"->Configuration Properties"->"Linker"->"Additional Dependencies" and add path to your tensorflow.lib as a next line.

Make sure that the tensorflow.dll is in Output Directory (by default, this is Debug\Release under your project's folder) or either in the directory contained by the %PATH% environment variable.

Here’s an example how prepare models

To generated the graph.pb file need takes a graph definition and a set of checkpoints and freezes them together into a single file.

Here’s an example how create tensorflow.lib file from tensorflow.dll for windows

Licensed under the MIT License