torch.cuda.manual_seed

torch.cuda.manual_seed(seed) [source]

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

Параметры

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

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

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