pytorch设置随机数种子

import random
import torch 

seed=random.randint(1,454645)
#为CPU设置种子用于生成随机数，以使得结果是确定的
torch.manual_seed(seed) 
#为当前GPU设置随机种子；
torch.cuda.manual_seed(seed)
#为所有的GPU设置种子。
torch.cuda.manual_seed_all(seed)
print(seed)