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