TensorFlow for Go

 go语言有着很好的潜力,最近TensorFlow提供了go语言的API.这些API特别适合加载Python中创建的模型，并在Go应用程序中执行它们。不过这些API还没有很稳定,目前也只是支持Linux,Mac OSX平台..

安装

本文以MacOSX安装为例

一 . Mac OS X上安装TensorFlow

安装TensorFlow有几种方式可以选择:

• virtualenv
• "native" pip
• Docker
• installing from sources
  之前我是Docker安装了,所以这里我选择第一种virtualenv方式,更多方式请看官网
  https://www.tensorflow.org/install/install_go

1 . 安装pip

$ sudo easy_install pip

已经安装过了可以忽略.py2,和py3冲突解决方法请百度

2 . 安装virtualenv

 $ sudo pip install --upgrade virtualenv 

3. 创建virtualenv环境

创建文件夹 :$ mkdir ~/tensorflow
如果你电脑装了python2.7,执行

 $ virtualenv --system-site-packages ~/tensorflow

如果你电脑装了python3.x,执行

$ virtualenv --system-site-packages -p python3 ~/tensorflow

4 . 激活virtualenv环境

$ source ~/tensorflow/bin/activate

你会发现你的终端变成了这个样子

(tensorflow)$ 

5 . 安装TensorFlow 和所有激活TensorFlow所需要的包

如果你电脑装了python2.7,执行

(tensorflow)$ pip install --upgrade tensorflow 

如果你电脑装了python3.x,执行

 (tensorflow)$ pip3 install --upgrade tensorflow 

附图:下载比较慢

这里写图片描述

二 . 安装TensorFlow for Go

前文说到的TensorFlow要安装好.

1. 下载并解压TensorFlow C库文件 到/usr/local/lib

执行以下命令:

$ 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.3.0.tar.gz" |
 sudo tar -C $TARGET_DIRECTORY -xz

第三条语句比较长..

2 . C库安装后,下载合适的包和依赖

执行

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

3 . 确认TensorFlow forGO 已经生效

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

示例

创建一个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())
}

运行 hello_tf.go

go run hello_tf.go


----------
运行结果:
Hello from TensorFlow version 1.3.0

同时运行结果中可能会产生一些警告,我们可以忽略.运行截图

这里写图片描述