search

twni2016 / OrganSegRSTN_PyTorch

PyTorch implementation of OrganSegRSTN - CVPR 2018
http://ccvl.jhu.edu/
MIT License
108 stars 33 forks source link

RuntimeError: CUDA out of memory. #3

Closed suoranxiu closed 5 years ago

suoranxiu commented 5 years ago

Raising error "RuntimeError: CUDA out of memory." when I run the training.py on my gtx1660 with 6 GB memory. I would appreciate it if you can give me some advice on the code of training steps .

twni2016 commented 5 years ago

Please use GPU with larger memory like 12GB. The batch size is 1, so the GPU memory cannot be reduced anymore.

suoranxiu commented 5 years ago

Thank you！

------------------ 原始邮件 ------------------ 发件人: "Tianwei Ni"notifications@github.com; 发送时间: 2019年5月6日(星期一) 中午1:27 收件人: "twni2016/OrganSegRSTN_PyTorch"OrganSegRSTN_PyTorch@noreply.github.com; 抄送: "索冉修"542663467@qq.com; "Author"author@noreply.github.com; 主题: Re: [twni2016/OrganSegRSTN_PyTorch] RuntimeError: CUDA out of memory.(#3)

Please use GPU with larger memory like 12GB. The batch size is 1, so the GPU memory cannot be reduced anymore.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.