Newbie question but I was not able to google an answer.
When running Keras model on GPU (with TensorFlow backend) a message is displayed automatically showing total and free amount of GPU memory.
Can I get the same functionality in PyTorch? How can I display the total and free GPU memory with PyTorch?
What’s the simplest way to display total GPU memory consumed by all Variables in my program?