0
点赞
收藏
分享

微信扫一扫

MacOS 安装C语言版TensorFlow

蓝莲听雨 2024-06-06 阅读 31

文章目录

安装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
# 在 Linux 和 macOS 上,您可能需要解压缩到 /usr/local/lib
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语言示例

  • 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编译器
gcc hello_tf.c -ltensorflow -o hello_tf
# 执行
./hello_tf
  • 输出结果

在这里插入图片描述

举报

相关推荐

0 条评论