torch.cuda.seed_all

torch.cuda.seed_all() [source]

Устанавливает seed для генерации случайных чисел на случайное число на всех GPU. Безопасно вызывать эту функцию, если CUDA недоступен; в этом случае она будет проигнорирована.