TensorFlow in Go

Construct and execute TensorFlow graphs in Go.

WARNING: The API defined in this package is not stable and can change without notice. The same goes for the awkward package path (github.com/tensorflow/tensorflow/tensorflow/go).

Quickstart

Refer to Installing TensorFlow for Go

Building the TensorFlow C library from source

If the "Quickstart" instructions above do not work (perhaps the release archives are not available for your operating system or architecture, or you're using a different version of CUDA/cuDNN), then the TensorFlow C library must be built from source.

Prerequisites

• bazel

• Environment to build TensorFlow from source code (Linux or OS X). If you don't need GPU support, then try the following:

  sudo apt-get install python swig python-numpy # Linux
  brew install swig                             # OS X with homebrew
  

Build

1. Download the source code

   go get -d github.com/tensorflow/tensorflow/tensorflow/go
   

2. Build the TensorFlow C library:

   cd ${GOPATH}/src/github.com/tensorflow/tensorflow
   ./configure
   bazel build --config opt //tensorflow:libtensorflow.so
   

   This can take a while (tens of minutes, more if also building for GPU).

3. Make libtensorflow.so available to the linker. This can be done by either:

   a. Copying it to a system location, e.g.,

   sudo cp ${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow/libtensorflow.so /usr/local/lib
   

   OR

   b. Setting environment variables:

   export LIBRARY_PATH=${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow
   # Linux
   export LD_LIBRARY_PATH=${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow
   # OS X
   export DYLD_LIBRARY_PATH=${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow
   

4. Build and test:

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

Generate wrapper functions for ops

Go functions corresponding to TensorFlow operations are generated in op/wrappers.go. To regenerate them:

Prerequisites:

• Protocol buffer compiler (protoc) 3.x
• The TensorFlow repository under GOPATH

go generate github.com/tensorflow/tensorflow/tensorflow/go/op

Support

Use stackoverflow and/or Github issues.

Contributions

Contributions are welcome. If making any signification changes, probably best to discuss on a Github issue before investing too much time. Github pull requests are used for contributions.