How to rotate a 3d image stack

Hi there,
I have an image stack in tiff format,in shape of (depth* widthheight). I want to rotate it around axis 0 . I tried ndimage.rotate, but it runs too slow as my image has a large size, 6001200*1200 typically. So I turn to Pytorch as it has cuda backend. How can I do my rotation with pytorch? I really need help. Thanks!