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.

Sign up for GitHub

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

Jump to bottom

Can this be trained using GPU? #35

Open
qazwsdc opened this issue May 24, 2019 · 6 comments
Open

Can this be trained using GPU? #35

qazwsdc opened this issue May 24, 2019 · 6 comments

Comments

@qazwsdc
Copy link

qazwsdc commented May 24, 2019

It's very slow using CPU to train a model
@yu4u
Copy link
Owner

yu4u commented May 24, 2019

You can train the model using a GPU simply by following README.

@qazwsdc
Copy link
Author

qazwsdc commented May 24, 2019

I looked closely but didn't find any information about GPU.

@qazwsdc
Copy link
Author

qazwsdc commented May 24, 2019

Could you please give some suggestions?
Thank you

@yu4u
Copy link
Owner

yu4u commented May 24, 2019

Keras automatically uses GPU if it is available.

@qazwsdc
Copy link
Author

qazwsdc commented May 25, 2019

I use RTX 2080ti .I have followed your README step but it's totally not use GPU at all.
Maybe should I install some cuda drivers?
My system is Windows 10.

@yu4u
Copy link
Owner

yu4u commented May 25, 2019

You might have to install tensorflow-gpu.
Anyway, I think this is not the problem of this repository, but the environment of the Keras framework.
Thus please refer to different documentations for appropriate environment construction.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@yu4u @qazwsdc