pytorch固定随机种子

在 PyTorch 中，可以使用以下代码来固定随机种子：

import torch

# 设置随机种子
torch.manual_seed(0)
torch.cuda.manual_seed(0)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

其中，`torch.manual_seed()` 和 `torch.cuda.manual_seed()` 分别用于设置 CPU 和 GPU 的随机种子；`torch.backends.cudnn.deterministic` 设置为 True 可以保证每次运行结果相同，而 `torch.backends.cudnn.benchmark` 设置为 False 可以避免一些不必要的计算开销。