TensorFlow.js for Node currently supports the following platforms:
The platforms listed include Mac OS X, Linux, and Windows operating systems, and support is provided for both CPU and GPU environments. Developers who want to use TensorFlow.js for Node on these platforms will need to ensure that their system meets the specified requirements.
- 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 11.2 w/ CUDNN v8) (see installation instructions)
- Windows CPU (Windows 7 or higher)
- Windows GPU (Windows 7 or higher and Cuda 11.2 w/ CUDNN v8) (see installation instructions)
For GPU support, [email protected] or later requires the developers to have certain versions of NVIDIA software installed on their system. The specific software requirements are listed in the table below.
Name | Version |
---|---|
NVIDIA® GPU drivers | CUDA 11.2 |
CUDA® Toolkit | 11.2 |
cuDNN SDK | 8.1.0 |
Other Linux variants might also work but this project matches core TensorFlow installation requirements.
npm install @tensorflow/tfjs-node
or
yarn add @tensorflow/tfjs-node
npm install @tensorflow/tfjs-node-gpu
or
yarn add @tensorflow/tfjs-node-gpu
Adopted from:
- https://github.com/SashiDo/content-moderation-image-api
- https://github.com/infinitered/nsfwjs
Preview: https://api.mindustry.me