标签: javascript node.js tensorflowjs
我试图在使用节点包 tfjs-node-gpu 开发的GPU上使用运行ML代码。当我尝试这样做时,程序正在运行,但是看不到任何进程正在添加到GPU。我怀疑出于这个原因我的程序未使用GPU。我刚刚安装了该软件包,并在引用它的代码中使用了它。这是正确的方法吗,我需要做更多的事情才能使TF node.js代码在GPU上运行。
在我的代码开始执行之前,我看到以下错误
以下是我的程序运行时的GPU日志
TIA