A.1. Installing tfjs-node-gpu on Linux

  1. We assume you have installed Node.js and npm on your system and that the paths to node and npm are included in your system path. If not, see https://nodejs.org/en/download/ for downloadable installers.
  2. Download the CUDA Toolkit from https://developer.nvidia.com/cuda-downloads. Be sure to choose the suitable version for the version of tfjs-node-gpu you intend to use. At the time of this writing, the latest version of tfjs-node-gpu is 1.2.10, which works with CUDA Toolkit version 10.0. In addition, be sure to select the correct operating system (Linux), architecture (for example, x86_64 for machines with mainstream Intel CPUs), Linux distribution, and version of the distribution. You will have the option ...

Get Deep Learning with JavaScript now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.