Is there any solution get computation resource?

I’m currently trying to profile the computation resource usage in PyTorch. While searching for information, I found that TensorFlow Lite has a benchmark tool that can measure CPU frequency, available memory, and other metrics. Is there a similar method or library available in PyTorch to obtain these metrics?