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, version 10.12.6 (Sierra) ou ultérieure
Configurer
Bibliothèque TensorFlow 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.