tfjs-node-gpu not using GPU

Asked Jun 20 '19 at 13:12

Active Jun 20 '19 at 13:12

Viewed 304 times

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

asked Jun 20 '19 at 13:12

Sai Raghuram Kaligotla

Giving us your code could help – Sebastian Speitel Jun 20 '19 at 13:21
Did you try uninstalling and installing node-gpu again? – Sebastian Speitel Jun 20 '19 at 13:22
Yes I have tried doing that , but did not help. – Sai Raghuram Kaligotla Jun 20 '19 at 13:24
And no errors or warings during installation? Using the current version? – Sebastian Speitel Jun 20 '19 at 13:25
No errors while installing but got the above posted error while executing.Do I need to do anything in the code other than importing node-gpu? – Sai Raghuram Kaligotla Jun 20 '19 at 13:26
Not that I know of, just `const tfjs = require('node-gpu')` – Sebastian Speitel Jun 20 '19 at 13:27
These are my actual import statements `const tf = require("@tensorflow/tfjs-node");` `const dicomParser = require('dicom-parser');` – Sai Raghuram Kaligotla Jun 20 '19 at 13:28
That's just the node version not the cpu nor the gpu one, you need `'@tensorflow/tfjs-node-gpu'` – Sebastian Speitel Jun 20 '19 at 14:21
sorry I was using the same '@tensorflow/tfjs-node-gpu' . Missed it in the comment – Sai Raghuram Kaligotla Jun 20 '19 at 14:23
@SebastianSpeitel any idea about the issue? – Sai Raghuram Kaligotla Jun 25 '19 at 14:50
I have solved this myself , that was totally problem with NVIDIA drivers , uninstalling and installing them solved my issue. – Sai Raghuram Kaligotla Jun 27 '19 at 10:10

tfjs-node-gpu not using GPU

0 Answers0

Linked