CIT icon indicating copy to clipboard operation
CIT copied to clipboard

Published 20 hours ago • verified publisher icon Amazingren

"RuntimeError: CUDA error: out of memory" when train TOM with RTX2080Ti

Open kris-yangjs opened this issue 2 years ago • 2 comments

Hello, thank you for your amazing work! I tried to train the model, and I finished the GMM step. But when I tried to train the TOM with one GPU(2080Ti), I got a error: "RuntimeError: CUDA error: out of memory " , even if I changed the batch size to 1. And what I want to ask is : what kind of GPU you used to train ? 2022-03-29 14-50-10 的屏幕截图

kris-yangjs avatar Mar 29 '22 06:03 kris-yangjs

same question

pearzzz avatar Dec 14 '22 02:12 pearzzz

V100 GPU(32G)

ovipaul avatar May 25 '23 08:05 ovipaul