RuntimeError: cuda runtime error (2) : out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58

Open Tomingz opened this issue 7 years ago • 3 comments

ubuntu16.04 ++940mx(2gb)+cuda8.0+py27+pytorch0.3.0post4 .I have changed batch_test=1 in option.py .but it is always out of memory,how to solve it?

Dec 13 '18 01:12 Tomingz

You can change a computer 😅 2gb GPU is too small

Dec 19 '18 13:12 SMZCC

@SMZCC I think too

Dec 21 '18 05:12 Tomingz

I have solved this problem by changing the batch in option.py

Dec 21 '18 06:12 Tomingz