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