torch.cuda.manual_seed_all
torch.cuda.manual_seed_all(seed)[source]-
Sets the seed for generating random numbers on all GPUs. It’s safe to call this function if CUDA is not available; in that case, it is silently ignored.
- Parameters
-
seed (int) – The desired seed.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.cuda.manual_seed_all.html