Google is committed to advancing racial equity for Black communities. See how.

Installer TensorFlow pour Go

TensorFlow fournit une API Go particulièrement utile pour charger des modèles créés avec Python et les exécuter dans une application Go.

Plates-formes compatibles

TensorFlow pour Go est compatible avec les systèmes d'exploitation suivants :

  • Linux (64 bits et x86)
  • macOS X, version 10.12.6 (Sierra) ou ultérieure

Configurer

Bibliothèque Tensorflow pour C

Installez la bibliothèque TensorFlow C, qui est requise pour le package TensorFlow Go.

Télécharger

Téléchargez et installez le package TensorFlow Go et ses dépendances :

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

Ensuite, validez l'installation :

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

Créer

Exemple de programme

Une fois le package TensorFlow Go installé, créez un exemple de programme avec le code source suivant (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())
}

Exécuter

Exécutez l'exemple de programme suivant :

go run hello_tf.go

La commande devrait renvoyer le message suivant : Hello from TensorFlow version number

Si le programme génère les messages d'avertissement suivants, vous pouvez les ignorer :

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.

Compiler à partir de la source

TensorFlow est une technologie Open Source. Consultez les instructions pour compiler TensorFlow pour Go à partir du code source.