adds list_gpu_processes function (#44616)
Summary:
per title, to make it easier to track the creation of stray contexts:
```
python -c "import torch; a=torch.randn(1, device='cuda'); print(torch.cuda.memory.list_gpu_processes(0)); print(torch.cuda.memory.list_gpu_processes(1))"
GPU:0
process 79749 uses 601.000 MB GPU memory
GPU:1
no processes are running
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44616
Reviewed By: mruberry
Differential Revision: D23675739
Pulled By: ngimel
fbshipit-source-id: ffa14cad9d7144e883de13b1c2c6817bd432f53a
Author
Natalia Gimelshein