-
Notifications
You must be signed in to change notification settings - Fork 470
CUDA out of memory with batchsize = 1 in one 1080Ti #34
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
Comments
I also tried running CornerNet on a 1080 Ti. I was able to train the network with a batch size of 3. Can you give more details about your settings? |
Thank you for your reply, and we finally find it's due to our Arch system made some compile errors to pyTorch(didn't open the cudnn). After we recompile the pytorch, it works! |
I got the same question with you, but I correctly installed cudnn and it works fine in other code. |
Have you solve the problem? I also installed cudnn correctly but it doesn't work and still out of memory with the batchsize = 1. Why??? Thank you! |
No description provided.
The text was updated successfully, but these errors were encountered: