100 lines (67 sloc) 3.04 KB

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 package path: (


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.


  • bazel

  • Environment to build TensorFlow from source code (Linux of macOS). 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


  1. Download the source code

    go get -d
  2. Build the TensorFlow C library:

    cd ${GOPATH}/src/
    bazel build -c opt //

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

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

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

    sudo cp ${GOPATH}/src/ /usr/local/lib


    b. Setting environment variables:

    export LIBRARY_PATH=${GOPATH}/src/
    # Linux
    export LD_LIBRARY_PATH=${GOPATH}/src/
    # OS X
    export DYLD_LIBRARY_PATH=${GOPATH}/src/
  4. Build and test:

    go test

Generate wrapper functions for ops

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


go generate


Use stackoverflow and/or Github issues.


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.