npm i @tensorflow/tfjs-node-gpu Repository github.com/tensorflow/tfjs Homepage github.com/tensorflow/tfjs#readme Weekly Downloads 2,291 Version 4.22.0 License Apache-2.0 Unpacked Size 2.02 MB Total Files 569 Issues 320 Pull Requests 256 Last publish 2 months ago Collaborators Try on RunKit Re...
Using tfjs-node-gpu on a GKE cluster running on an n1-higmem-8 with an NVIDIA P4 or V100 GPU fails when the cuda_malloc_async allocater is set using TF_GPU_ALLOCATOR. System information Have I written custom code (as opposed to using a s...
4.22.0•Public• Published5 months ago For information please referhere. Readme Keywords none Install npm i@tensorflow/tfjs-node-gpu Repository github.com/tensorflow/tfjs Weekly Downloads 1,920 License Apache-2.0 Unpacked Size 2.02 MB
(node:28032) UnhandledPromiseRejectionWarning: Error: Invalid TF_Status: 13 Message: Blas xGEMM launch failed : a.shape=[1,1,4], b.shape=[1,4,12], m=1, n=12, k=4 at NodeJSKernelBackend.executeSingleOutput (/<localdir>/node_modules/@tensorflow/tfjs-node-gpu/dist/nodejs_kernel_...
import * as tf from '@tensorflow/tfjs-node';is not valid for node servers, my project is not a create-react-app type client-app. "dependencies":{"@tensorflow/tfjs-node-gpu":"^0.3.2","lodash":"^4.17.11","shuffle-seed":"^1.1.6"}} ...