javascript - tfjs-node-gpu 不使用 GPU

问题描述

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

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

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

TIA

标签： javascriptnode.jstensorflowjs

javascript - tfjs-node-gpu 不使用 GPU

问题描述

解决方案

推荐阅读