-
Notifications
You must be signed in to change notification settings - Fork 2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Trying hard to install tfjs-node-gpu on Windows 10 #1589
Comments
Can you try deleting your https://github.com/tensorflow/tfjs-node/blob/master/WINDOWS_TROUBLESHOOTING.md |
@nkreeger yes I followed those. I made sure the I'm still getting:
|
What I did and it worked for
Let it fail. Go to node_modules folder. And run |
However, my plan is to run training with the GPU, to have 10x speeds. But if I apply the same strategy there here's what I get:
|
@theodorDiaconu, does 0.3.2 install? |
@adwellj I managed to be able to install that, by doing Still with 1.1.2 node-gyp:
|
I am now happily using 0.3.2. The experience to get it working was absolutely awful. Still I was expecting better improvements |
@theodorDiaconu, glad to hear you got it working! Agreed, GPU installation is a very poor experience at this time. Fingers crossed that with enough feedback the team will be able to fix the GPU installations for the new versions. For future reference, I didn't have to apply any special flags or go back and do any rebuilds. I simply install Regarding the GPU performance, I'm going from 550 sec/epoch with CPU to 103 sec/epoch with GPU. I would imagine that some models will see a larger benefit with GPU than others. For instance: #468 (comment) |
Understood, indeed I had a small model. I'm still learning about this. Very glad that I know when I have to do serious training I can use my GPU. This ticket should still be open. Building doesn't work on ^0.3.2 |
With the latest tfjs-node/tfjs-node-gpu package, we are shipping pre-compile addon and there should be no compilation on your machine. This issue should have been fixed. |
Using python2.7
Installed windows utilities
Stuck here:
The text was updated successfully, but these errors were encountered: