-
Notifications
You must be signed in to change notification settings - Fork 217
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
RuntimeError: CUDA out of memory #27
Comments
What kind of GPU do you have ? |
thanks for your help! I reduced the images to 128*128, now I can set batch size to 16 (if higher than 16 it still goes wrong) |
It depends on your GPU and the C3D implementation. How many parameters are you training ? |
I have same problem @MounirB . I use my own dataset and set begin_frame = 1,end_frame = 500,size = 256,342,number of classes = 5. |
hello there!
I was running the UCF101_3DCNN.py in pycharm, and this issue showed up:
RuntimeError: CUDA out of memory. Tried to allocate 1.19 GiB (GPU 0; 6.00 GiB total capacity; 3.60 GiB already allocated; 738.91 MiB free; 3.60 GiB reserved in total by PyTorch)
I changed the batch size to 1 and it disappeared. but I guess it is not a proper way to solve the problem. hope someone could offer some suggestion.
The text was updated successfully, but these errors were encountered: