文档章节

安装TensorFlow for Go

片刻
 片刻
发布于 2017/06/13 21:45
字数 807
阅读 130
收藏 1

 

 

原文链接 : https://www.tensorflow.org/install/install_go

译文链接 : http://www.apache.wiki/pages/viewpage.action?pageId=10029605

贡献者 : 片刻 ApacheCN Apache中文网

TensorFlow提供在Go程序中使用的API。这些API特别适合加载在Python中创建的模型,并在Go应用程序中执行它们。本指南介绍如何安装和设置 TensorFlow Go软件包

警告: TensorFlow API API 不在 TensorFlow API稳定性保证范围内

 

支持的平台

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

  • Linux的
  • Mac OS X

 

安装

TensorFlow for Go取决于TensorFlow C库。采取以下步骤安装此库并启用TensorFlow for Go:

  1. 决定是否只在CPU或GPU的帮助下运行TensorFlow for Go。为了帮助您决定,请阅读以下指南之一标题为“确定要安装的TensorFlow”的部分。

  2.  /usr/local/lib通过调用以下shell命令,下载并提取TensorFlow C库:

    TF_TYPE="cpu" # Change to "gpu" for GPU support

    TARGET_DIRECTORY='/usr/local'

    curl -L \

      "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-$(go env GOOS)-x86_64-1.1.0.tar.gz" |

    sudo tar -C $TARGET_DIRECTORY -xz 

    tar命令将TensorFlow C库提取到lib 子目录中TARGET_DIRECTORY。例如,指定/usr/localTARGET_DIRECTORY使tar提取TensorFlow C库成/usr/local/lib

    如果您希望将库提取到不同的目录中,请进行相应的调整TARGET_DIRECTORY

  3. 在步骤2中,如果您指定了系统目录(例如/usr/localTARGET_DIRECTORY,则运行ldconfig以配置链接器。例如:

    sudo ldconfig 

    如果您分配了TARGET_DIRECTORY系统目录(例如,~/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 

  4. 现在安装了TensorFlow C库,请go get按如下方式调用相应的软件包及其依赖关系:

    go get github.com/tensorflow/tensorflow/tensorflow/go

      

  5. 调用go test如下来验证TensorFlow for Go安装:

    go test github.com/tensorflow/tensorflow/tensorflow/go

    去测试github.com/tensorflow/tensorflow/tensorflow/go

     

如果go getgo test生成错误消息,搜索(或发布到) StackOverflow 可能的解决方案。

 

Hello World

安装TensorFlow对围棋后,输入以下代码到一个文件名为hello_tf.go

package main

 

import (

    tf "github.com/tensorflow/tensorflow/tensorflow/go"

    "github.com/tensorflow/tensorflow/tensorflow/go/op"

    "fmt"

)

 

func main() {

    // Construct a graph with an operation that produces a string constant.

    s := op.NewScope()

    c := op.Const(s, "Hello from TensorFlow version " + tf.Version())

    graph, err := s.Finalize()

    if err != nil {

        panic(err)

    }

 

    // Execute the graph in a session.

    sess, err := tf.NewSession(graph, nil)

    if err != nil {

        panic(err)

    }

    output, err := sess.Run(nil, []tf.Output{c}, nil)

    if err != nil {

        panic(err)

    }

    fmt.Println(output[0].Value())

有关Go中TensorFlow的更高级示例,请查看API文档中的 示例,该文档使用预先训练过的TensorFlow模型来标记图像内容。

运行

 hello_tf.go通过调用以下命令运行:

go run hello_tf.go

Hello from TensorFlow version number 

该程序还可能会生成以下窗体的多个警告消息,您可以忽略它们:

W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library

wasn't compiled to use *Type* instructions, but these are available on your

machine and could speed up CPU computations. 

从源代码构建

TensorFlow是开源的。您可以从TensorFlow源代码中构建TensorFlow for Go,并按照 单独文档中的说明进行操作

© 著作权归作者所有

共有 人打赏支持
片刻
粉丝 98
博文 268
码字总数 306559
作品 0
海淀
高级程序员
私信 提问
TensorFlow——快速安装

TensorFlow:2015年Google开源的机器学习框架 1、Anoconda安装 (1)Window,MacOS,Linux都已支持Tensorflow。 (2)Window用户只能使用python3.5(64bit)。MacOS,Linux支持python2.7和python3...

飞天小橘子
04/19
0
0
TensorFlow平台的搭建

一、TensorFlow介绍 TensorFlow™ 是一个采用数据流图(data flow graphs),用于数值计算的开源软件库。节点(Nodes)在图中表示数学操作,图中的线(edges)则表示在节点间相互联系的多维数...

qq_39525832
05/28
0
0
第3章 TensorFlow入门

第3章 TensorFlow入门 TensorFlow是Google创造的数值运算库,作为深度学习的底层使用。本章包括: TensorFlow介绍 如何用TensorFlow定义、编译并运算表达式 如何寻求帮助 注意:TensorFlow暂...

优惠券发放
05/31
0
0
TensorFlow集成TensorRT环境配置

本文前提是cuda和cudnn以及TensorRT已经安装完毕,具体详情,可以参考上一篇文章: https://developer.nvidia-china.com/forum.php?mod=viewthread&tid=8767&extra=page%3D1 TensorRT下载地址...

AI科技大本营
04/26
0
0
安装适用于 Java 的 TensorFlow

安装适用于 Java 的 TensorFlow TensorFlow 可提供在 Java 程序中使用的 API。这些 API 特别适合用于加载以 Python 语言创建的模型并在 Java 应用中运行这些模型。本指南将介绍如何安装适用于...

程序员诗人
06/06
0
0

没有更多内容

加载失败,请刷新页面

加载更多

微服务分布式事务实现

https://www.processon.com/view/link/5b2144d7e4b001a14d3d2d30

WALK_MAN
今天
2
0
《大漠烟尘》读书笔记及读后感文章3700字

《大漠烟尘》读书笔记及读后感文章3700字: 在这个浮躁的社会里,你有多久没有好好读完一本书了? 我们总觉得自己和别人不一样,所以当看到别人身上的问题时,很少有“反求诸己”,反思自己。...

原创小博客
今天
3
0
大数据教程(9.5)用MR实现sql中的jion逻辑

上一篇博客讲解了使用jar -jar的方式来运行提交MR程序,以及通过修改YarnRunner的源码来实现MR的windows开发环境提交到集群的方式。本篇博主将分享sql中常见的join操作。 一、需求 订单数据表...

em_aaron
今天
3
0
十万个为什么之什么是resultful规范

起源 越来越多的人开始意识到,网站即软件,而且是一种新型的软件。这种"互联网软件"采用客户端/服务器模式,建立在分布式体系上,通过互联网通信,具有高延时(high latency)、高并发等特点...

尾生
今天
3
0
Terraform配置文件(Terraform configuration)

Terraform配置文件 翻译自Terraform Configuration Terraform用文本文件来描述设备、设置变量。这些文件被称为Terraform配置文件,以.tf结尾。这一部分将讲述Terraform配置文件的加载与格式。...

buddie
今天
5
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部