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:
(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
Build
-
Download the source code
go get -d github.com/tensorflow/tensorflow/tensorflow/go
-
Build the TensorFlow C library:
cd ${GOPATH}/src/github.com/tensorflow/tensorflow
./configure
bazel build -c opt //tensorflow:libtensorflow.so
This can take a while (tens of minutes, more if also building for GPU).
-
Make libtensorflow.so
and libtensorflow_framework.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
sudo cp ${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow/libtensorflow_framework.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
-
Build and test:
go generate github.com/tensorflow/tensorflow/tensorflow/go/op
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:
go generate github.com/tensorflow/tensorflow/tensorflow/go/op
Support
Use Stack Overflow
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.