TensorFlow在文件c_api.h
中定义的C语言API,可用于构建其他语言的绑定。
支持的平台
您可以在以下操作系统上安装TensorFlow for C:
安装
以下步骤用于安装TensorFlow for C:
-
确定是否需要GPU支持,参阅“确定要安装的TensorFlow”的部分。
-
下载并提取TensorFlow C库
/usr/local/lib
, 使用以下shell命令:TF_TYPE="cpu" # Change to "gpu" for GPU support OS="linux" # Change to "darwin" for Mac OS TARGET_DIRECTORY="/usr/local" curl -L \ "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-${OS}-x86_64-1.3.0.tar.gz" | sudo tar -C $TARGET_DIRECTORY -xz
tar
命令将TensorFlow C库提取到lib
的子目录TARGET_DIRECTORY
。 -
在步骤2中,如果指定了
系统
目录(例如,/usr/local
)作为TARGET_DIRECTORY
,那么可以直接运行ldconfig
配置链接器。例如:sudo ldconfig
如果指定的不是系统目录(例如,
~/mydir
),那么你必须附加提取目录(例如,~/mydir/lib
)到两个环境变量。例如:export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib # For both Linux and Mac OS X export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/mydir/lib # For Linux only export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:~/mydir/lib # For Mac OS X only
验证您的安装
安装TensorFlow for 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;
}
构建和运行
编译hello_tf.c
,命令如下:
gcc hello_tf.c
运行生成的可执行文件应该输出以下消息:
a.out Hello from TensorFlow C library version number
故障排除
如果构建程序失败,最可能的原因是gcc
找不到TensorFlow C库。解决这个问题的一个方法是给gcc
指定-I
和-L
选项。例如,如果TARGET_LIBRARY
是/usr/local
目录,那么使用gcc
如下:
gcc -I/usr/local/include -L/usr/local/lib hello_tf.c -ltensorflow
如果执行a.out
失败了,可能是以下问题:
- 程序编译有没有错误?
- 是否将正确的目录分配给步骤3中指出的环境变量安装?
- 是否导出了这些环境变量?