torch.cuda.seed

torch.cuda.seed() [source]

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

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

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