Skip to content

Instantly share code, notes, and snippets.

@korakot
Last active November 22, 2022 08:43
Show Gist options
  • Star 3 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save korakot/22abd6eccac229e9cb9a027b088b50d6 to your computer and use it in GitHub Desktop.
Save korakot/22abd6eccac229e9cb9a027b088b50d6 to your computer and use it in GitHub Desktop.
TFJS node.ipynb
Display the source blob
Display the rendered blob
Raw
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
var run_async = async (pf) => {
$$.async();
await pf();
$$.done();
};
// in case you want tensorflow
sh('npm install @tensorflow/tfjs-node-gpu')
var tf = require('@tensorflow/tfjs-node-gpu');
console.log(tf.backend().isUsingGpuDevice);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment