February 2020
Intermediate to advanced
560 pages
22h 18m
English
To use the GPU-accelerated version of TensorFlow.js (tfjs-node-gpu) in Node.js, you need to have CUDA and CuDNN installed on your machine. First of all, the machine should be equipped with a CUDA-enabled NVIDIA GPU. To check whether the GPU in your machine meets that requirement, visit https://developer.nvidia.com/cuda-gpus.
Next, we list the detailed steps of the driver and library installation for Linux and Windows, as these are the two operating systems on which tfjs-node-gpu is currently supported.
Read now
Unlock full access