有没有一种简单的方法来计算在GPU中运行的pytorch代码？

Question

在GPU（而非CPU）中运行时，有没有简单的方法来计算pytorch代码（python）？我可以简单地使用python中内置的time模块来计算CPU中的代码。但是有没有简单的方法来使用函数来为GPU计时代码？感谢。