torch.cuda.seed_all

torch.cuda.seed_all() [исходный код]

Задаёт начальное значение для генерации случайных чисел, используя случайное число на всех графических процессорах.

Эту функцию можно безопасно вызывать, если CUDA недоступна; в этом случае вызов игнорируется без уведомления.