torch.cuda.manual_seed_all
torch.cuda.manual_seed_all(seed)
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.