Instalar o TensorFlow para Go

O TensorFlow fornece uma API Go, especialmente útil para carregar modelos criados com Python e executá-los em um aplicativo Go.

Plataformas compatíveis

O TensorFlow para Go é compatível com os seguintes sistemas:

  • Linux, 64-bit, x86
  • macOS, versão 10.12.6 (Sierra) ou mais recente

Configuração

Biblioteca C do TensorFlow

Instale a biblioteca C do TensorFlow, necessária para o pacote do TensorFlow Go.

Fazer o download

Faça o download e instale o pacote do TensorFlow Go e as respectivas dependências:

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

Depois, valide a instalação:

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

Criar

Programa de exemplo

Com o pacote TensorFlow Go instalado, crie um programa de exemplo com o seguinte código-fonte (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())
}

Executar

Execute o programa de exemplo:

go run hello_tf.go

O comando gera: Hello from TensorFlow version number

Talvez o programa gere as seguintes mensagens de aviso, que você pode ignorar:

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.

Criar da origem

O TensorFlow é de código aberto. Leia as instruções para criar o TensorFlow para Go a partir do código-fonte.