Skip to content
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

tensorflow-gpu not working. #3

Open
dathayer23 opened this issue Jan 31, 2018 · 2 comments
Open

tensorflow-gpu not working. #3

dathayer23 opened this issue Jan 31, 2018 · 2 comments

Comments

@dathayer23
Copy link

I have installed tensorflow-gpu on my desktop which has two high powered NVidia gpu cards and it unfortunately is not using the gpu. I have implicitly told it to import tensorflow-gpu and it causes an error. Does anyone know how to get the main.py code to use the GPU?

@dathayer23
Copy link
Author

BTW have installed tensorflow-gpu on my anaconda tensorflow instance. Everything installed successfully.

@bourdakos1
Copy link
Owner

bourdakos1 commented Feb 15, 2018

Hey, sorry it took me so long to see this.
If you have 2 gpus you should use the dist-version otherwise it will only use one of the gpus. You shouldn't have to specifically import tensorflow-gpu in the code though. Do you know if it ran on any of your cards?

You should see something like this with you card listed:

Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: Tesla P100 ...)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants