Nv-nsight-cu-cli Does not work in NGC Pytorch container 20.03

When I try to use nsight compute in the NGC container 20.03 on pytorch scripts, I get the following error.
free(): invalid pointer.
The profiler works on simple *.cu files but not on pytorch sripts.

It’s a known issue.
We’ve isolated, fixed it, and will release the fixed version in the next container release.

OK That’s great thanks. BTW may I know what was causing it, Because I tried to install pytorch from source in the container and it also didn’t work. However I installed a nightly pytroch binary and it worked.
For my purposes I actually need to make some changes in pytorch source and use that version. But this issue prevents me to do that. Is there any way?