tfjs-node-gpu不使用GPU

时间:2019-06-20 13:12:44

标签: javascript node.js tensorflowjs

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

在我的代码开始执行之前,我看到以下错误

enter image description here

以下是我的程序运行时的GPU日志

enter image description here

TIA

0 个答案:

没有答案