torch.cuda.manual_seed — PyTorch 2.7 documentation (original) (raw)

torch.cuda.manual_seed(seed)[source][source]¶

Set the seed for generating random numbers for the current GPU.

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.

Warning

If you are working with a multi-GPU model, this function is insufficient to get determinism. To seed all GPUs, use manual_seed_all().

• torch.cuda.manual_seed
  • manual_seed()

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.