Unfortunately, modules can save arbitrary elements during forward. So it is really hard to have a common api that computes this.
If torchsummary does not work, you most likely will have to run a forward and check the actual consumption.