-
Notifications
You must be signed in to change notification settings - Fork 9
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
CUDA out of memory #8
Comments
At least 18G memory is required for training phase, our models are trained on 24G Titan X graphical cards. |
thank you |
But it's said that there is 6.6G consumed with 1200*1600 images input in the paper. |
It's test stage
…------------------ 原始邮件 ------------------
发件人: ***@***.***>;
发送时间: 2021年5月18日(星期二) 晚上6:49
收件人: ***@***.***>;
抄送: ***@***.***>; ***@***.***>;
主题: Re: [yhw-yhw/D2HC-RMVSNet] CUDA out of memory (#8)
At least 18G memory is required for training phase, our models are trained on 24G Titan X graphical cards.
If you only have 12G 1080Ti graphical cards, I suggest you to decrease the settings for training, e.g. number of plane sweep, number of neighbor views in train.sh.
But it's said that there is 6.6G consumed with 1200*1600 images input in the paper.
—
You are receiving this because you authored the thread.
Reply to this email directly, view it on GitHub, or unsubscribe.
|
Thanks, how much memory consumed if batchsize is 1 |
1080Ti * 4, batch_size = 4
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 10.92 GiB total capacity; 10.09 GiB already allocated; 5.00 MiB free; 19.09 MiB cached)
The text was updated successfully, but these errors were encountered: