-
Notifications
You must be signed in to change notification settings - Fork 65
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
About the module train #32
Comments
If you have a compatible GPU, Keras should automatically train on it. |
Thank you very much for your help |
Sorry, now the code can run on GPU, but the computer memory occupation rate increases gradually with the code running until it crashes. I can't solve this problem. Is there any way to effectively solve this problem? |
Hello, can you tell me how to train with gpu? I have been groping for a day without solving this problem. My Python version is 3.7 and cuda version is 11.2. What version should I use for tensorflow and keras? Is an updated version required? Do I need to change the code after updating the version? Thank you very much for your help!!! |
|
Thank you! I will try it |
Sorry to bother you again. I also encountered this problem. How did you solve it at that time? Thank you very much for your help. |
Hello!Do you train your model on GPU? During training, I found that the code only runs on the CPU. Now I want to put the code on the GPU. Is there any way?
The text was updated successfully, but these errors were encountered: