tfjs-node-gpu not using GPU

泪湿孤枕 提交于 2021-01-29 05:15:04

问题


I am trying to use run ML code on GPU developed using the node package tfjs-node-gpu . When I try doing so the program is running but I am unable to see any process being added to the GPU. I suspect that my program is not using GPU for this reason. I have just installed the package and used in the code referring it. Is this the right way do I need to do anything more to make my TF node.js code run on GPU.

I am seeing tthe below error before my code starts executing

And the below is the GPU logs when my program is running

TIA

来源:https://stackoverflow.com/questions/56686767/tfjs-node-gpu-not-using-gpu

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!