首页 > 解决方案 > Brain.js 没有使用 GPU 进行训练

问题描述

我一直在使用 Brain.js 来训练神经网络,它工作得很好,只是它似乎只使用 CPU 来训练神经网络。

我正在使用 Windows,任务管理器显示使用 ~25% CPU 的节点进程(我假设它正在最大化单个线程)。查看 MSI Afterburner,GPU 根本没有被使用。

GPU是Nvidia RTX 2060 super。

我该怎么做才能让 Brain.js 使用我的 GPU?我已经四处搜索,但到目前为止还没有找到太多信息......

标签: node.jsneural-networkgpucpubrain.js

解决方案


推荐阅读