TensorFlow backend for TensorFlow.js via Node.js
This repo is under active development and is not production-ready. We are actively developing as an open source project.
TensorFlow.js for Node currently supports the following platforms:
- Mac OS X CPU (10.12.6 Siera or higher)
- Linux CPU (Ubuntu 14.04 or higher)
- Linux GPU (Ubuntu 14.04 or higher and Cuda 9.0 w/ CUDNN v7) (see installation instructions)
- Windows 7 or later (Currently, CPU only)
Other Linux variants might also work but this project matches core TensorFlow installation requirements.
Installing CPU TensorFlow.js for Node:
npm install @tensorflow/tfjs-nodeyarn add @tensorflow/tfjs-node
Installing Linux GPU TensorFlow.js for Node:
npm install @tensorflow/tfjs-node-gpuyarn add @tensorflow/tfjs-node-gpu
Mac OS X Requires Xcode
If you do not have Xcode setup on your machine, please run the following commands:
After that operation completes, re-run
yarn add or
npm install for the
Using the binding
Before executing any TensorFlow.js code, import the node package:
;// Load the binding;// Or if running with GPU:;
# Download and install JS dependencies, including libtensorflow 1.8.yarn# Run TFJS tests against Node.js backend:yarn test
# Switch to GPU for local development:yarn enable-gpu
MNIST demo for Node.js
See the tfjs-examples repository for training the MNIST dataset using the Node.js bindings.
Optional: Build libtensorflow From TensorFlow source
This requires installing bazel first.
bazel build --config=monolithic //tensorflow/tools/lib_package:libtensorflow