GPU Out of Memory

Question

GPU Out of Memory

DNHanXu opened this issue 6 years ago · 3 comments

I made some changes on original codes to make it compatible with torch 1.0.1 and python 3.6. However, I encounter gpu out of memory problem during testing process. Does anyone have the same problem? And is it caused by version difference? I used 1080ti with 11GB memory.

Answer 1 · 2019-05-23T05:55:10.000Z

make some changes according to version update of Pytorch from 0.3 to 0.4 will solve this problem

Answer 2 · 2019-07-07T04:30:49.000Z

I had same problem,could you tell me how to solve it?

Answer 3 · 2019-07-08T02:12:31.000Z

@usename6 For torch >= 0.4, try to use 'torch.no_grad()' instead of 'volatile' flag.