保证生成数据一致性
device调整为gpu生成
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
torch.cuda.manual_seed(0)
print(torch.rand(1, device=device))
保证生成数据一致性
device调整为gpu生成
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
torch.cuda.manual_seed(0)
print(torch.rand(1, device=device))