torch.cuda.seed

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

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

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

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

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