From here, it shows ‘You can use torch.manual_seed()
to seed the RNG for all devices (both CPU and CUDA)’. Does it mean torch.manual_seed()
is also for GPU?
However, from here, it shows that torch.cuda.manual_seed(seed)
and torch.cuda.manual_seed_all(seed)
are needed to set seed for GPU. I feel a little confused now.