torch.cuda.manual_seed

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

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

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

Параметры:

seed (int) – Желаемое начальное значение.

Предупреждение

Если вы работаете с моделью, использующей несколько графических процессоров, этой функции недостаточно для обеспечения детерминированности. Чтобы задать начальные значения для всех графических процессоров, используйте manual_seed_all().