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)
pytorch设置随机数种子
最新推荐文章于 2024-07-24 15:22:35 发布