torch.cuda.manual_seed_all

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

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

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

Параметры:

seed (int) – Требуемое начальное значение.