Функция torch.cuda.seed_all в PyTorch
Справочное руководство Pytorch torch
torch.cuda.seed_allЭто функция в PyTorch, используемая для установки начального числа генератора случайных чисел для всех устройств CUDA. Устанавливает начальное число генератора случайных чисел для всех устройств CUDA (функционально идентична manual_seed_all).
Определение функции
torch.cuda.seed_all()
Пример использования
Пример
import torch
# Установить начальное число генератора случайных чисел для всех устройств CUDA
if torch.cuda.is_available():
torch.cuda.seed_all()
x = torch.randn(3, 3).cuda()
print(f"Случайный тензор: {x}")
print(f"Количество устройств CUDA: {torch.cuda.device_count()}")
else:
print("CUDA недоступна")
# Установить начальное число генератора случайных чисел для всех устройств CUDA
if torch.cuda.is_available():
torch.cuda.seed_all()
x = torch.randn(3, 3).cuda()
print(f"Случайный тензор: {x}")
print(f"Количество устройств CUDA: {torch.cuda.device_count()}")
else:
print("CUDA недоступна")
Другие расширения