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.
6 Likes