文章目录
安装C语言版TensorFlow
- 官方文档:https://tensorflow.google.cn/install/lang_c?hl=zh-cn
- TensorFlow 提供了一个 C API,该 API 可用于为其他语言构建绑定。该 API 在 c_api.h 中定义,旨在实现简洁性和一致性,而不是便利性。
| TensorFlow C 库 | 网址 |
|---|
| Linux | |
| Linux(仅支持 CPU) | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz |
| Linux(支持 GPU) | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz |
| macOS | |
| macOS(仅支持 CPU) | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-2.15.0.tar.gz |
| Windows | |
| Windows(仅支持 CPU) | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-windows-x86_64-2.6.0.zip |
| Windows(仅支持 GPU) | https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-windows-x86_64-2.6.0.zip |
解压归档
sudo rm -rf /usr/local/LICENSE
sudo rm -rf /usr/local/THIRD_PARTY_TF_C_LICENSES
sudo rm -rf /usr/local/lib/libtensorflow*
sudo rm -rf /usr/local/include/tensorflow
sudo tar -C /usr/local -xzf libtensorflow-cpu-darwin-x86_64-2.15.0.tar.gz
sudo chown -R $(whoami):$(whoami) /usr/local
sudo chmod -R u+w /usr/local
环境变量
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/lib
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:/usr/local/lib
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib
c_api.h
- https://github.com/tensorflow/tensorflow/blob/master/tensorflow/c/c_api.h

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 -ltensorflow -o hello_tf
./hello_tf
