mirror of
https://github.com/tensorflow/haskell.git
synced 2024-11-23 03:19:44 +01:00
Haskell bindings for TensorFlow
eb7e78d60d
I'm not sure why, but in some cases it seems linking only works if *both* the .so and the .dylib are present in /usr/local/lib. This may be due to a quirk of how Bazel builds the library, and/or how ghc/stack load the library. |
||
---|---|---|
ci_build | ||
docker | ||
docs/haddock | ||
google-shim | ||
tensorflow | ||
tensorflow-core-ops | ||
tensorflow-mnist | ||
tensorflow-mnist-input-data | ||
tensorflow-nn | ||
tensorflow-opgen | ||
tensorflow-ops | ||
tensorflow-proto | ||
tensorflow-queue | ||
tensorflow-test | ||
third_party | ||
tools | ||
.gitignore | ||
.gitmodules | ||
CONTRIBUTING.md | ||
LICENSE | ||
README.md | ||
stack.yaml |
The tensorflow-haskell package provides Haskell bindings to TensorFlow.
This is not an official Google product.
Instructions
Build with Docker on Linux
As an expedient we use docker for building. Once you have docker working, the following commands will compile and run the tests.
git clone --recursive https://github.com/tensorflow/haskell.git tensorflow-haskell
cd tensorflow-haskell
IMAGE_NAME=tensorflow/haskell:v0
docker build -t $IMAGE_NAME docker
# TODO: move the setup step to the docker script.
stack --docker --docker-image=$IMAGE_NAME setup
stack --docker --docker-image=$IMAGE_NAME test
There is also a demo application:
cd tensorflow-mnist
stack --docker --docker-image=$IMAGE_NAME build --exec Main
Build on Mac OS X
The following instructions were verified with Mac OS X El Capitan.
-
Install dependencies via Homebrew:
brew install swig brew install bazel
-
Build the TensorFlow library and install it on your machine:
cd third_party/tensorflow ./configure # Choose the defaults when prompted bazel build -c opt tensorflow:libtensorflow_c.so install bazel-bin/tensorflow/libtensorflow_c.so /usr/local/lib ln -s /usr/local/lib/libtensorflow_c.{so,dylib} cd ../..
-
Run stack:
stack test
Note: you may need to upgrade your version of Clang if you get an error like the following:
tensorflow/core/ops/ctc_ops.cc:60:7: error: return type 'tensorflow::Status' must match previous return type 'const ::tensorflow::Status' when lambda expression has unspecified explicit return type
return Status::OK();
In that case you can just upgrade XCode and then run gcc --version
to get the new version of the compiler.