I’m trying to bottleneck my code.
The code is able to run under cProfiler and autograd cpu profiler. However it will raise out of memory error when running under cuda profiler.
And there’s no error stack; I got this:
terminate called after throwing an instance of ‘std::runtime_error’
what():  /opt/pytorch/torch/csrc/autograd/profiler.h:53: out of memory
BTW: I also monitored the gpu memory usage: the memory usage was shown to be 800mb while the whole ram size is 11g.