当前位置: 首页>>技术教程>>正文


安装TensorFlow C语言版

qingchuan 技术教程 , , 去评论

TensorFlow在文件c_api.h中定义的C语言API,可用于构建其他语言的绑定

支持的平台

您可以在以下操作系统上安装TensorFlow for C:

  • Linux
  • Mac OS X
  • 安装

    以下步骤用于安装TensorFlow for C:

    1. 确定是否需要GPU支持,参阅“确定要安装的TensorFlow”的部分。

    2. 下载并提取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

    3. 在步骤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中指出的环境变量安装
    • 是否导出了这些环境变量?

    参考资料

    本文由《纯净的天空》出品。文章地址: https://vimsky.com/article/3625.html,未经允许,请勿转载。