Error when loading tensorflow.js with node.js

Question

I am fairly new to javascript. I am working through the tensorflow-js tutorial on using tensorflow-js with node.js however when I run the example code:

const tf = require('@tensorflow/tfjs');

// Load the binding:
require('@tensorflow/tfjs-node');  // Use '@tensorflow/tfjs-node-gpu' if running with GPU.

// Train a simple model:
const model = tf.sequential();
model.add(tf.layers.dense({units: 100, activation: 'relu', inputShape: [10]}));
model.add(tf.layers.dense({units: 1, activation: 'linear'}));
model.compile({optimizer: 'sgd', loss: 'meanSquaredError'});

const xs = tf.randomNormal([100, 10]);
const ys = tf.randomNormal([100, 1]);

model.fit(xs, ys, {
  epochs: 100,
  callbacks: {
    onEpochEnd: async (epoch, log) => {
      console.log(`Epoch ${epoch}: loss = ${log.loss}`);
    }
  }
});

I have already installed the tensorflowjs package: npm install @tensorflow/tfjs-node-gpu

However I get the error:

module.js:549
throw err;
Error: Cannot find module '@tensorflow/tfjs-node'

I am not sure what is causing this error.

Geuis · Accepted Answer · 2018-07-13 20:49:37Z

7

You need to:

npm install @tensorflow/tfjs-node

edited Jul 13, 2018 at 20:49

answered Jul 13, 2018 at 19:56

Geuis

42.5k57 gold badges164 silver badges223 bronze badges

Sign up to request clarification or add additional context in comments.

5 Comments

piccolo Over a year ago

This worked! Would you mind adding a small comment about what this command does - as it is not listed in the tensorflowjs installation instructions

Geuis Over a year ago

Its right there in the installation instructions github.com/tensorflow/…

piccolo Over a year ago

it does not mention --save

Geuis Over a year ago

Right, you probably don't need that if installing from an existing package.json

Hawkeye64 Over a year ago

// Load the binding: require('@tensorflow/tfjs-node'); // Use '@tensorflow/tfjs-node-gpu' if running with GPU. Right there in the comments of the code. You can use @tensorflow/tfjs-node or if using a GPU with CUDA you can use @tensorflow/tfjs-node-gpu - install the approriate one. They are mutually exclusive.

Collectives™ on Stack Overflow

Error when loading tensorflow.js with node.js

1 Answer 1

5 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

5 Comments

Your Answer

Sign up or log in

Post as a guest

Related