Is there a way to get an overview of which tensors and models are on which GPUs?
Ideally would be a function that returns a list of tensors/models given the cuda device index.
It would be very handy when debugging multi-GPU training scripts.
Thanks in advance,