安装 C 版 TensorFlow

TensorFlow 提供了一个 C API,该 API 可用于为其他语言构建绑定。该 API 在 c_api.h 中定义,旨在实现简洁性和一致性,而不是便利性。

支持的平台

以下系统支持 C 版 TensorFlow:

  • Linux - 64 位,x86
  • macOS X - 版本 10.12.6 (Sierra) 或更高版本

设置

下载

TensorFlow C 库网址
Linux
Linux(仅支持 CPU) https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.12.0.tar.gz
Linux(支持 GPU) https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-1.12.0.tar.gz
macOS
macOS(仅支持 CPU) https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-1.12.0.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

如果程序无法编译,请确保 gcc 可以访问 TensorFlow C 库。如果解压缩到 /usr/local,请将库位置显式传递给编译器:

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

从源代码编译

TensorFlow 是开源系统。请阅读相关说明,了解如何从源代码编译 TensorFlow 的 C 库。