如何在C语言中使用tensorflow（三）测试

发表于 更新于 分类于 阅读次数：

本文主要介绍tensorflow的cpu版本的C语言使用方法。

下载libtensorflow对应版本后，可以测试libtensorflow在当前环境下是否可用。

测试代码路径：

DIR：

|--lib:
    |--libtensorflow.so
    |--libtensorflow_framework.so

|--include:
    |--/include/tensorflow/c/c_api.h
    |--/include/tensorflow/c/LICENSE

|--src:
    |--test.c
    |--Makefile

测试代码test.c：

1
2
3
4
5
6
7
#include <stdio.h>
#include <tensorflow/c/c_api.h>

int main() {
  printf("Hello from TensorFlow C library version %s\n", TF_Version());
  return 0;
}

Makefile文件:

1
2
3
4
5
6
7
8
9
10
INC=-I../include
LIB=-L../Lib \
        -ltensorflow \
	-ltensorflow_framework 
SRC=test.c
CC=gcc
all:
	${CC}  ${SRC} ${INC} ${LIB} -o test
clean:
	rm test

编译过程：

1
2
make     #编译生成可执行文件
make clean     #删除可执行文件

运行可执行文件test：

1
2
./test
Hello from TensorFlow C library version 1.15.0

至此，动态库版本与硬件版本匹配完成。