Always out of memory, but available GPU memory is 16GB!
binbinmeng opened this issue · 1 comments
binbinmeng commented
RuntimeError: cuda runtime error (2) : out of memory at /opt/pytorch/pytorch/torch/lib/THC/generic/THCStorage.cu:58
Yijunmaverick commented
@binbinmeng I modified the test.py by adding volatile=True and find that it can test large image size (e.g., 1024) now. Have a try.