RuntimeError: CUDA out of memory

Open onnx20 opened this issue 6 years ago • 2 comments

ubuntu16.04+cuda10+pytorch1.1.1+NVIDIA2080

I have a problem, the "RuntimeError: CUDA out of memory. Tried to allocate 52.00 MiB (GPU 0; 7.76 GiB total capacity; 5.63 GiB already allocated; 29.69 MiB free; 82.78 MiB cached)"

Thanks

Aug 07 '19 08:08 onnx20

Hi Try to use with torch.no_grad(): to inference. You can refer to https://github.com/vlad3996/FaceDetection-DSFD

Aug 07 '19 12:08 swordlidev

@lijiannuist Can you clarify where to put torch.no_grad() in demo.py? I am getting the same error.

Sep 12 '19 15:09 nyck33