Using @tensorflow/tfjs-node-gpu 3.0.0 and CUDA 10.0 on Ubuntu 20.04
// Prepares and returns 4GB data async function prepareData () { return { inputs: inputT