How to get GPU memory usage in pytorch code?

Is there any way to see the gpu memory usage in pytorch code?

2 Likes

You also have all the functions to get the memory allocated and the memory actually used by tensors. See the doc here for more details.

5 Likes