How to ensure the repeatability of the results of the Pytorch model across devices？

Question

I am a little bit confused about the randomness of the pytorch model.

I used the following code to fix the random seed so that the training results of the model can be re