Skip to content

Latest commit

 

History

History
107 lines (77 loc) · 3.71 KB

lang_c.md

File metadata and controls

107 lines (77 loc) · 3.71 KB
Raw

安装 TensorFlow C API

TensorFlow 提供了用于构建绑定其他语言的 C API。这些 API 定义在 c_api.h 中，并且优先考虑其设计得简洁与一致而非便利。

支持平台

TensorFlow C API 支持以下系统:

  • Linux，64-bit，x86
  • macOS X，10.12.6 版本（Sierra）或更高

步骤

下载

TensorFlow C 函数库 URL
Linux
Linux 下仅支持 CPU https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.10.1.tar.gz
Linux 下支持 GPU https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-1.10.1.tar.gz
macOS
macOS 下仅支持 CPU https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-1.10.1.tar.gz

提取

将下载的 TensorFlow C 函数库提取到 /usr/local/lib 下（或使用另一目录）：

sudo tar -xz libtensorflow.tar.gz -C /usr/local

链接器

如果你将 TensorFlow C 函数库提取到一个系统目录下，比如 /usr/local，使用 ldconfig 设置链接器：

sudo ldconfig

或者，如果你将 TensorFlow C 函数库提取到一个非系统目录下，比如 ~/mydir，那设置链接器环境变量：

Linux

export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/mydir/lib

mac OS

export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:~/mydir/lib

构建

示例程序

安装完成 TensorFlow C 函数库后，按照以下源代码创建一个示例程序（hello_tf.c）：

#include <stdio.h>
#include <tensorflow/c/c_api.h>

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

编译

将示例程序编译创建出可执行文件，然后运行：

gcc hello_tf.c -o hello_tf

./hello_tf

这个命令会输出：Hello from TensorFlow C library version number

成功：TensorFlow C 函数库配置完成。

如果程序没能够构建成功，确定 gcc 可以访问 TensorFlow C 函数库。如果提取到 /usr/local，请将函数的库的目录地址传递给编译器：

gcc -I/usr/local/include -L/usr/local/lib hello_tf.c -ltensorflow -o hello_tf

由源码构建

TensorFlow 已开源。查看说明{:.external}了解如何由源码构建 TensorFlow C 函数库。